torchvision.datasets 中数据集的 .data 属性不适用于 ImageFolder？

Question

我可以像这样访问 MNIST 对象的训练数据集：

from torchvision.datasets import MNIST
trainset = MNIST(root='.', train=True, download=True)
print(trainset.data)

但是，我无法使用 ImageFolder 做同样的事情。有人知道 ImageFolder 的等价物吗？

AttributeError: 'ImageFolder' 对象没有属性 'data'

Answer 1

没有。

如今，每个人都假设 MNIST 适合内存，因此 it is preloaded to the data attribute。但是，这对于 assets 来说通常是不可能的。因此，images are loaded on-the-fly 表示它们没有 ImageDataset 属性。您可以使用 data 或 self.imgs 访问图像路径和标签。