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

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

原文链接：https://blog.csdn.net/u010682375/article/details/77746489

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

相关文章

热门资讯