tensorflow基础之：图片地址和标签的CSV文件的创建

最新推荐文章于 2020-04-15 15:19:51 发布

原创

最新推荐文章于 2020-04-15 15:19:51 发布 · 1.5k 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #csv创建

本文介绍如何为TensorFlow项目创建包含图片地址和标签的CSV文件。首先在项目目录建立IMAGE文件夹，存储有命名规则的图片。接着通过代码读取图片列表，并生成CSV，将每张图片的路径和对应标签写入。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、概念
无论是图片分类还是目标追踪，都需要提供每张图片对应的类别或者标注。这篇文章主要介绍一下如何创建图片对应的类别的csv文件。
二、思路
首先在项目目录下创建一个名为IMAGE文件夹，将一些图片放入其中，图片名称最好有固定的命名规则（如图1）。之后通过代码获得图片路径，将所有图片名称读取到列表中，创建csv文件，利用循环结构，一次写入一张图片的地址和标签。
图1，图片
三、代码实现

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