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

您所在的位置：网站首页 › 如何批量生成文件夹名 › 如何用python生成带图片名称和标签的.txt文件（代码）

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

2024-07-08 02:17| 来源: 网络整理| 查看: 265

我们之前实现了如何用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.记得把

outer_path = '/home/lincanran/cls_exp/place100'

改成自己的文件夹目录的地址，然后保存；；

3.在终端输入：

python turn_to_list_many.py