如何用python生成带图片名称和标签的.txt文件（代码）

我们之前实现了如何用python批量修改图片的名称，不清楚的同学可以看一下这一篇：

python批量修改一个文件夹下含多个文件夹中的所有图片名称（代码）

 

接下来我们来看一下如何生成带图片名称和标签的txt文件

因为我们在用caffe进行分类训练时，不管是生成imdb还是直接拿图片训练，都是需要标签文件的

话不多说，直接上代码：

#!/usr/bin/python
# -*- coding:utf-8 -*-

import os
def generate(dir,label):
	files = os.listdir(dir)
	files.sort()
	print '****************'
	print 'input :',dir
	print 'start...'
	listText = open('all_list.txt','a')
	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 '****************'


outer_path = '/home/lincanran/cls_exp/place100'   #这里是你的图片的目录


if __name__ == '__main__':
	i = 0
	folderlist = os.listdir(outer_path)          #列举文件夹
	for folder in folderlist:
		generate(os.path.join(outer_path, folder),i)
		i += 1

 

具体操作：

1.新建一个  turn_to_list_many.py  文件，然后把上面的代码复制进来，然后保存。

2.在终端输入：

python turn_to_list_many.py 

 

生成成功，我们可以在目录下看到多了一个 all_list.txt文件：

 

打开该文件：

 

每一行是一个样本，然后用一个空格隔开，后面是它的标签

我这边一共是100个类（0-99的标签），然后每个类1500张图片，所以总数是150000，没有问题！！！