python实现对文件中图片生成带标签的txt文件方法
在深度学习中经常需要生成带标签的图片名称列表,xxxlist.txt文件,下面写一个简单的python脚本生成该文件列表。
importos
defgenerate(dir,label):
files=os.listdir(dir)
files.sort()
print'****************'
print'input:',dir
print'start...'
listText=open(dir+'\\'+'list.txt','w')
forfileinfiles:
fileType=os.path.split(file)
iffileType[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实现对文件中图片生成带标签的txt文件方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。