深度学习标签制作简易Python代码

最新推荐文章于 2024-08-01 17:17:39 发布

Magic_Anthony

最新推荐文章于 2024-08-01 17:17:39 发布

阅读量2.9k

点赞数 2

分类专栏：图像处理 PYTHON 深度学习文章标签：深度学习 Python 数据集标签

本文链接：https://blog.csdn.net/magic_anthony/article/details/80596978

版权

本文介绍了一个简单的3步方法来制作深度学习训练数据集的标签，包括在Excel中填充数据和标签，保存为CSV文件，然后使用Python读取CSV文件并映射到数据集中的标签。这种方法适合初学者，对于有扎实Python基础的开发者，可能更倾向于使用如TensorFlow的TFRecord等高级方法。

摘要由CSDN通过智能技术生成

如果你想制作一个原始数据到标签一一映射的训练数据集，这里给你一个简易的想法，大概分3步。

①在Excel表中填入你的数据和标签，以下分别是原始图像和Excel表：

②将Excel表保存成.csv格式，另存为--》浏览--》.csv：

③通过.csv文件中的信息，分别调用你数据集中的数据，并返回对应标签,Python代码如下：

#Anthony
#本程序作用是通过读取CSV文件中的信息如‘image011’，其中‘image01’为图片的名称，‘1’为图像标签。
#然后通过CSV文件中的信息来调用文件‘文件路径+image01’来调取图像。

import csv
import matplotlib.pyplot as plt # plt 用于显示图像
import matplotlib.image as mpimg