图片数据转换为csv

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')