pytorch读取自制数据集

方法一

数据集中数据按照类别，进行文件放置。

import torch
import torchvision
import torchvision.transforms as transforms

# 读取数据集
datasets = torchvision.datasets.ImageFolder(root="数据集路径",
												transform=transforms.Compose([
												&#