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

在深度学习中经常需要生成带标签的图片名称列表，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实现对文件中图片生成带标签的txt文件方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

• 利用Python批量生成任意尺寸的图片
• Python利用matplotlib生成图片背景及图例透明的效果
• 从零学python系列之教你如何根据图片生成字符画
• python生成验证码图片代码分享
• python使用pil生成图片验证码的方法
• 利用Python爬取微博数据生成词云图片实例代码
• python实现基于两张图片生成圆角图标效果的方法
• Python生成数字图片代码分享
• python生成圆形图片的方法
• 利用Python如何生成便签图片详解