pytorch创建本身的数据集Dataset

源码下载，分别在label.py、test.py 将图片集（下图class）存到在当前工程目录下，并按类别分开（下图的一、二、3）： 为图片生成label：python import os path = "class" # 图片集路径 classes = [i for i in os.listdir(path)] files = os.listdir(path) train = open("t

>>阅读原文<<