如何将MNIST图像加载到Pytorch DataLoader中？

109

特里：

pytorch的数据加载和处理教程非常具体地针对一个示例，有人可以帮我了解更通用的简单图像加载功能应该是什么样的吗？

教程：http：//pytorch.org/tutorials/beginner/data_loading_tutorial.html

我的资料：

我在以下文件夹结构中将MINST数据集作为jpg包含在其中。（我知道我可以只使用数据集类，但这纯粹是为了了解如何在不使用csv或复杂功能的情况下将简单图像加载到pytorch中）。

文件夹名称是标签，图像是灰度的28x28 png，无需任何转换。

杜安：

这是我为pytorch 0.4.1做的（仍应在1.3中使用）

def load_dataset():
    data_path = 'data/train/'
    train_dataset = torchvision.datasets.ImageFolder(
        root=data_path,
        transform=torchvision.transforms.ToTensor()
    )
    train_loader = torch.utils.data.DataLoader(
        train_dataset,
        batch_size=64,
        num_workers=0,
        shuffle=True
    )
    return train_loader

for batch_idx, (data, target) in enumerate(load_dataset()):
    #train network

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。