如何使用python获取图像datasets的标签

最新推荐文章于 2024-08-15 08:00:00 发布

爱不持久

最新推荐文章于 2024-08-15 08:00:00 发布

阅读量1.4k

点赞数 1

分类专栏：深度学习卷积神经网络 python相关学习文章标签： python 深度学习人工智能机器学习

本文链接：https://blog.csdn.net/wacebb/article/details/107720748

版权

python相关学习同时被 3 个专栏收录

42 篇文章 5 订阅

订阅专栏

深度学习

32 篇文章 3 订阅

订阅专栏

卷积神经网络

26 篇文章 0 订阅

订阅专栏

在我们训练神经网络的时候，需要有标签的数据，通过以下方法获取：

数据文件夹格式如下：

test下有三个类别：1 2 3

import os

path = "F:/python/pytorch/vgg/data2/test"  # 图片集路径
classes = [i for i in os.listdir(path)]
files = os.listdir(path)
test = open("test.txt", 'w')
for i in classes:
    s = 0
    for imgname in os.listdir(os.path.join(path, i)):
        name = os.path.join(path, i) + '\\' + imgname + '   ' + str(classes.index(i)) + '\n'
        test.write(name)
        s += 1

test.close()

只需要修改路径，以及test等名称，生成文件在工程目录下。

结果如下：包括路径和类别。