代码如下:
def mkdir_if_not_exist(path):
#如果path存在,返回true,如果path不存在,返回false
if not os.path.exists(os.path.join(*path)):
os.makedirs(os.path.join(*path))#可生成多层递归目录
mkdir_if_not_exist(['H:/阿里云/data/train_valid'])
#读取图片的路径和标签
task='skirt_length_lables'
#热身数据与训练数据的图片标记文件
warmup_label_dir='H:/阿里云/data/web/Annotations/skirt_length_labels.csv'
base_label_dir='H:/阿里云/data/base/Annotations/label.csv'
image_path=[]
with open(warmup_label_dir,'r')as f:
#readlines将文件中数据放入整个list中
lines=f.readlines()
#split分隔符,通过
tokens=[l.rstrip().split(',')for l in lines]
for path,_,label in tokens:
image_path.append(('data/web/'+path,label))
#训练集
with open(base_label_dir,'r')as f:
lines=f.readlines()
tokens=[l.rstrip().split(',')for l in lines]
for path,tk,label in tokens:
image_path.append(('data/base/'+path,label))
image_path
#如果path存在,返回true,如果path不存在,返回false
if not os.path.exists(os.path.join(*path)):
os.makedirs(os.path.join(*path))#可生成多层递归目录
mkdir_if_not_exist(['H:/阿里云/data/train_valid'])
#读取图片的路径和标签
task='skirt_length_lables'
#热身数据与训练数据的图片标记文件
warmup_label_dir='H:/阿里云/data/web/Annotations/skirt_length_labels.csv'
base_label_dir='H:/阿里云/data/base/Annotations/label.csv'
image_path=[]
with open(warmup_label_dir,'r')as f:
#readlines将文件中数据放入整个list中
lines=f.readlines()
#split分隔符,通过
tokens=[l.rstrip().split(',')for l in lines]
for path,_,label in tokens:
image_path.append(('data/web/'+path,label))
#训练集
with open(base_label_dir,'r')as f:
lines=f.readlines()
tokens=[l.rstrip().split(',')for l in lines]
for path,tk,label in tokens:
image_path.append(('data/base/'+path,label))
image_path
输出结果为:
文件路径加标签