python实现对文件中图片生成带标签的txt文件方法

import osdef generate(dir,label):  files = os.listdir(dir)  files.sort()  print '****************'  print 'input :',dir  print 'start...'  listText = open(dir+'//'+'list.txt','w')  for file in files:    fileType = os.path.split(file)    if fileType[1] == '.txt':      continue        name = file + ' ' + str(int(label)) +'/n'    listText.write(name)  listText.close()  print 'down!'  print '****************'  if __name__ == '__main__':   generate('C://Users//Desktop//image',1)