您的位置:首页 > 编程语言 > Python开发

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

2017-08-31 14:52 941 查看
  在深度学习中经常需要生成带标签的图片名称列表,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.
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  python