import os
import csv
def create_csv(dirname):
path='./chongqing1_round1_testA_20191223/'+dirname+'/'
name=os.listdir(path)
with open (dirname+'.csv','w') as csvfile:
writer=csv.writer(csvfile)
writer.writerow(['data','label'])
for n in name:
if n[-4:]=='.jpg':
print(n)
writer.writerow(['./chongqing1_round1_testA_20191223/'+str(dirname)+'/'+str(n),'./chongqing1_round1_testA_20191223/' + str(dirname) + 'label/' + str(n[:-4] + '.png')])
else:
pass
if __name__ == "__main__":
create_csv('images')
图片数据转换为csv
最新推荐文章于 2024-08-23 09:38:05 发布