遍历文件夹写入txt
将一级目录下,二级目录内的图片名字读成标签数据形式的txt文件
# coding=UTF-8
import os
import sys
path="/home/cdli/Original_10.18"
color_list=os.listdir(path)
l1=len(color_list)-1
i=0
with open("train_color.txt","w") as f:
for c_l in color_list:
path_2=path+"/"+c_l+"/"
pic=os.listdir(path_2)
print(c_l+','+str(i))
for p in pic:
f.write('/home/cdli/Original_10.18/'+c_l+'/'+p+' '+str(i)+'\n')
i=i+1