文件目录的格式如下：201-220 里面全是 图片
from torchvision.datasets import ImageFolder
from torchvision import transforms
import torch
import torch.nn as nntransform=transforms.Compose([transforms.RandomCrop(180), # 随机裁剪 这句话删除了 就会报错 transforms.RandomHorizontalFlip(), # 左右翻转transforms.ToTensor(), #将图片转换为Tensor,归一化至[0,1]transforms.Normalize(mean=[.5,.5,.5],std=[.5,.5,.5]) # 均值方差归一化
])dataset=ImageFolder('samples_test_support/train',transform=transform)# 图片存储的根目录，即各类别文件夹所在目录的上一级目录print(dataset.classes) #根据分的文件夹的名字来确定的类别
print(dataset.class_to_idx) #按顺序为这些类别定义索引为0,1...
print(dataset.imgs[0:10]) #返回从所有文件夹中得到的图片的路径以及其类别