1.只需要使用dataset.ImageFolder函数,就可以读取大文件夹里的小文件夹中的图片数据集
2.使用.imgs就可以读取具体图片数据集的路径
3.使用.class_to_index就可以读取子文件夹的标签数值
data_path = 'E:/1.DataSet/Tongue/DownloadData/Tongue coating/0.Tongue coating pic'
# 读取数据
dataset_train = datasets.ImageFolder(data_path, transform)
print(dataset_train.imgs)
# 对应文件夹的label
print(dataset_train.class_to_idx)
transform:
transform = transforms.Compose([
transforms.Resize((128, 128)),transforms.ToTensor(),
transforms.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])])