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)) +''    listText.write(name)  listText.close()  print 'down!'  print '****************'  if __name__ == '__main__':   generate('C:\Users\Desktop\image',1)