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

最新推荐文章于 2023-11-02 11:35:34 发布

原创

最新推荐文章于 2023-11-02 11:35:34 发布 · 3.1k 阅读

16 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #Python #数据集标签

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

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

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

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

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

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

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