制作数据集（二）--为图片数据集打上标签并保存为txt文件

概要

由于我自己的项目缺少合适的数据集，所以我需要自己制作数据集，即我需要自己完成从用摄像头录制视频到制作图片数据集的整个过程。上一篇博客记录了我将视频数据剪辑为图片数据的过程。这一片博客旨在记录我为图片制作标签txt文件的过程。因为只是第一次简单的尝试，所以只记录了简单的0，1标签

我制作的标签txt文件：（图片文件夹/图片名+标签（0/1））

一、图片分好文件夹很重要

要想对不同类型的图片打上不同的标签，首先要将图片按照不同的类别分好文件夹，文件夹的层次如下：

dataset/images/e0/e1/

要打上0标签的图片全放到dataset/images/e0/文件夹中，要打上1标签的图篇全放到dataset/images/e1/文件夹中。（当然，你自己可以自己定义文件夹的名字的，但是建议你按上面的文件夹层次建立文件夹）。到时候形成的标签txt文件存放在dataset/文件夹中。即你运行完打标签的代码后会生成txt文件。如下：

dataset/all_list.txtimages/e0/e1/

二、打标签的代码具体实现过程及详解

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

https://blog.csdn.net/qq_38451119/article/details/81947403

代码附于下方：

import os
def generate(dir, label):files = os.listdir(dir) #os.listdir() 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表。files.sort()  #对文件或文件夹进行排序print('****************')print('input :', dir)print('start...')listText = open('F:/learning/detection/Pytorch_Retinaface-master/engagement_dataset/all_list.txt', 'a+')  #创建并打开一个txt文件，a+表示打开一个文件并追加内容for file in files:  #遍历文件夹中的文件fileType = os.path.split(file) #os.path.split（）返回文件的路径和文件名，【0】为路径，【1】为文件名if fileType[1] == '.txt':  #若文件名的后缀为txt,则继续遍历循环，否则退出循环continuename = folder+ '/' +file + ' ' + str(int(label)) + '\n'  #name 为文件路径和文件名+空格+label+换行listText.write(name)  #在创建的txt文件中写入namelistText.close() #关闭txt文件print('down!')print('****************')outer_path = 'F:\learning\detection\Pytorch_Retinaface-master\engagement_dataset\images'  # 这里是你的图片路径if __name__ == '__main__':  #主函数i = 0folderlist = os.listdir(outer_path)# 列举文件夹for folder in folderlist:  #遍历文件夹中的文件夹(若engagement文件夹中存在txt或py文件，则后面会报错）generate(os.path.join(outer_path, folder), i)#调用generate函数，函数中的参数为：（图片路径+文件夹名，标签号）i += 1

三、运行代码的过程中遇到的问题

在运行代码的过程当中，当我第二次运行代码的时候回出现无法识别目录...all_list.txt或者是无法识别目录....make_label.py的错误，好像是因为我把txt文件和py文件都放到了images/文件夹中去了
导致我在第一次运行完代码后，在我的images文件夹中生成了all_list文件，且我把制作标签的Python文件make_label.py也都存放在了,当我的代码在images中遍历e0/和e1/文件夹是，遍历到了txt和py文件，从而导致识别不了。所以我觉得数据集的文件夹的层次设置，以及txt文件的存放位置很重要。

仿照着Retinaface_pytorch项目中的FDDB数据集的文件夹的设置方式，以及 **参考：如何用python生成带图片名称和标签的.txt文件（代码）**的代码，完成了给我自己的图片数据集制作标签txt文件。