PyTorch图片数据载入方法

倦王2025-11-03 14:51

官网具体内容：pytorch官网

root 代表主目录，主目录下一级目录为类别目录，dataset可以在载入过程中进行一些transform操作。代码介绍了transform具体可以是怎么样子的结构。

python 复制代码

import torch
from torch.utils.data import DataLoader
from torchvision import datasets,transforms

data_transform = transforms.Compose(
    [
    transforms.Resize(size = (64,64)),
    transforms.RandomHorizontalFlip(p=0.5),
    transforms.ToTensor()
    ]
)

下面就是如何实例化一个datasets.ImageFolder()类的方式。类中间的参数决定了 root：载入何处的数据，transform对数据采用何种变化，target_transform对类标签有何处理方式

python 复制代码

from torchvision import datasets

train_data = datasets.ImageFolder(root = train_dir,
                                 transform = data_transform,
                                 target_transform = None)
test_data = datasets.ImageFolder(root = test_dir,
                                transform = data_transform)

print(f"Train data{train_data}\nTest data{test_data}")

最后再强调一下目录结构：