如何使用pytorch中的torchvision.ImageFolder进行图像数据加载？

本文共计1200个文字，预计阅读时间需要5分钟。

torchvision.datasets 模块包含以API形式提供的数据集类，如 __getitem__ 和 __len__，它们都是 torch.utils.data.Dataset 的子类。因此，这些数据集类可以直接通过 torch.utils.data.DataLoader 使用，以实现多线程（Python的多进程）操作。例如：torch.

torchvision.datasets

Datasets 拥有以下API:

__getitem__

__len__

Datasets都是 torch.utils.data.Dataset的子类，所以，他们也可以通过torch.utils.data.DataLoader使用多线程（python的多进程）。

举例说明：

torch.utils.data.DataLoader(coco_cap, batch_size=args.batchSize, shuffle=True, num_workers=args.nThreads)

在构造函数中，不同的数据集直接的构造函数会有些许不同，但是他们共同拥有 keyword 参数。

transform： 一个函数，原始图片作为输入，返回一个转换后的图片。

target_transform - 一个函数，输入为target，输出对其的转换。例子，输入的是图片标注的string，输出为word的索引。