一、概念 无论是图片分类还是目标追踪，都需要提供每张图片对应的类别或者标注。这篇文章主要介绍一下如何创建图片对应的类别的csv文件。 二、思路 首先在项目目录下创建一个名为IMAGE文件夹，将一些图片放入其中，图片名称最好有固定的命名规则（如图1）。之后通过代码获得图片路径，将所有图片名称读取到列表中，创建csv文件，利用循环结构，一次写入一张图片的地址和标签。 三、代码实现
import os path = 'IMAGE' # 该文件是刚刚在项目文件夹下创建的，如果不在同意目录下，需要写绝对路径 file_name = os.listdir(path) # 返回包含IMAGE文件夹下所有项目的列表，存储在file_name中 str_text = '' # 用于之后的for循环中一次处理一个图片，写入csv文件的信息（路径+标签） with open('train_list.csv', 'w') as fid : for index in range(len(file_name)): str_text = path + os.sep + file_name[index] + ',' + '1' +'\n' # 这里假设类别是1 # str_text = path + os.sep + file_name[index] + ',' + file_name[index].split('-')[0] +'\n'，这种是读取图片的名字作为类别 # os.sep 是斜杠“\” fd.write(str_text) fid.close()生成的csv文件如下图2
