在深度学习中经常需要生成带标签的图片名称列表，xxxlist.txt文件，下面写一个简单的python脚本生成该文件列表。
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)) +
'\n'
listText.write(name)
listText.close()
print 'down!'
print '****************'
if __name__ ==
'__main__':
generate(
'C:\\Users\\Desktop\\image',
1)
运行该脚本，会在image文件夹中生成一个list.txt文件，并且每张图片带有标签1.
