生成带标签的图片名列表

在深度学习中经常需要生成带标签的图片名称列表，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)) +'
'
        listText.write(name)
    listText.close()
    print 'down!'
    print '****************'    

if __name__ == '__main__':  
    generate('C:\Users\Desktop\image',1) 

运行该脚本，会在image文件夹中生成一个list.txt文件，并且每张图片带有标签1.

摘自：http://blog.csdn.net/u010682375/article/details/77746489