当我们自定义数据集时,有时数据集是以图片的形式分好类给出的
torchvision.datasets.ImageFolder(root: str, transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, loader: Callable[[str], Any] = <function default_loader>, is_valid_file: Optional[Callable[[str], bool]] = None)
from torchvision import datasets dataset = datasets.ImageFolder('dataset/pokemon/') #没有transform,先看看取得的原始图像数据 print(dataset.classes) #根据分的文件夹的名字来确定的类别 print(dataset.class_to_idx) #按顺序为这些类别定义索引为0,1... print(dataset.imgs) #返回从所有文件夹中得到的图片的路径以及其类别
Pytorch自定义数据集之torchvision.datasets.ImageFolder
于 2022-04-20 10:48:44 首次发布