torch 读取图片

可以使用PyTorch中的torchvision库来读取图片。具体步骤如下：

1. 导入torchvision库

import torchvision

2. 使用torchvision.datasets.ImageFolder类加载图片数据集

dataset = torchvision.datasets.ImageFolder(root='path/to/dataset')

其中，'path/to/dataset'是存放图片数据集的文件夹路径。

3. 使用torch.utils.data.DataLoader类将数据集转换为可迭代的数据加载器

dataloader = torch.utils.data.DataLoader(dataset, batch_size=batch_size, shuffle=True)

其中，batch_size是每个批次的图片数量，shuffle=True表示每次迭代时打乱数据集顺序。

4. 遍历数据加载器，读取图片数据

for images, labels in dataloader:
    # images是一个大小为(batch_size, 3, height, width)的张量，表示一个批次的图片数据
    # labels是一个大小为(batch_size,)的张量，表示一个批次的图片标签