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

import os
def 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)