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

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

数据文件夹格式如下：

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等名称，生成文件在工程目录下。

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