Python Pytorch:Can’；t使用ImageFolder加载图像

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

data_dir = './train_dog'          # directory structure is    
train_dog/image
dset = datasets.ImageFolder(data_dir, transform)
train_loader = torch.utils.data.DataLoader(dset, batch_size=128, shuffle=True)

然而，它似乎不起作用。因此，我检查了存储的数据，如下所示

print dset[0][0]

然后它只显示3个张量（大小64x64）

文件夹中有10000多个图像。为什么它不能存储所有数据？

您应该尝试以下方法：

print len(dset)

它表示数据集的大小，也称为图像文件的数量

dset[0]

表示数据集的（无序）第一个索引，其中

dset[0][0]

包含输入图像张量，

dset[0][1]

包含相应的标签或目标。

您应该尝试以下操作：

print len(dset)

它表示数据集的大小，也称为图像文件的数量

---

dset[0]

表示数据集的（无序）第一个索引，其中

dset[0][0]

包含输入图像张量，

dset[0][1]

包含相应的标签或目标。

谢谢！成功了。但它在mini_batch=x_u.size（）[0]AttributeError中显示了文件“gan_scratch.py”，第258行：“list”对象没有属性“size”x_u包含上面加载的图像，但为什么它们不是张量而是list？谢谢！成功了。但它在mini_batch=x_u.size（）[0]AttributeError中显示了文件“ganu scratch.py”，第258行：“list”对象没有属性“size”x_u包含上面加载的图像，但为什么它们不是张量而是list？你应该改进你的帖子，不清楚你在问什么。你应该改进帖子，不清楚你在问什么。