利用Python根据目录下文件生成图片列表
跟我这样做:
- 定义训练参数
- 定义需要用到的变量、列表等
- 遍历每一类别的图片,并记录每一类别的数量和总数量
- 将遍历的图片路径、对应的类别存储到第二步定义的列表中
- 乱序后将相关信息存储到json文件
Let’s do it!
import os
import random
train_parameters = {
"label_dict":{
}, # 标签信息
'class_dim':0, # 类别个数
"train_list_path": "/home/data/train.txt", #train.txt路径
"eval_list_path": "/home/data/eval.txt", #eval.txt路径
"readme_path": "/home/data/readme.json", #readme.json路径
}
def get_data_list(dataPath, train_list_path, eval_list_path):
class_infor = [] # 存放所有类别,如:dog, cat, car……
class_label = 0 # 标记类别标签
class_num = 0 # 每一类多少张图片
class_dim =