我在尝试使用“ImageFolder”来加载图像。

data_dir = './train_dog'          # 目录结构为    train_dog/imagedset = datasets.ImageFolder(data_dir, transform)train_loader = torch.utils.data.DataLoader(dset, batch_size=128, shuffle=True)

然而，这似乎不起作用。所以我检查了存储的数据如下

print dset[0][0]

然后它只显示了3个张量（大小为64×64）。

[torch.FloatTensor of size 3x64x64]

文件夹中有超过10,000张图像。为什么它不能存储所有数据？

回答：

你应该尝试这样做：

print len(dset)

这表示数据集的大小，也就是图像文件的数量。

dset[0]表示数据集的（打乱后的）第一个索引，其中dset[0][0]包含输入图像张量，dset[0][1]包含相应的标签或目标。