根据文件夹中的图片数据集生成.txt文件标签

---

前言

深度学习中的与图片有关的任务，如图像识别，图像分类，目标检测，图像分割等，在准备图片数据集时，图片的序号和标签往往是由.txt文件获取的。本文将介绍如何给文件夹中的图片数据集上标签并保存的.txt文件中。

---

数据集格式

以二分类任务为例，我的数据集目录是:

E:/c/images/001.bad_apple

E:/c/images/002.good_apple

其中，001.bad_apple文件夹中包含若干张图片和一个Readme文件。图片命名格式是0.bmp、1.bmp、2.bmp……；Readme文件是一个.txt文件，用于说名该文件夹下的一些相关信息。002.bad_apple中的文件命名和格式与001.bad_apple完全相同。

一些细节：生成的.txt文件中，每一行的格式是"序号+空格+类别标签"。其中序号从1开始，类别标签从1开始。

<