Pytorch:无法使用ImageFolder加载图像

时间:2017-11-29 12:38:41

标签: python machine-learning deep-learning pytorch

我正在尝试使用“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)。

[torch.FloatTensor of size 3x64x64]

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

1 个答案:

答案 0 :(得分:1)

你应该试试这个:

print len(dset)

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

dset[0]表示数据集的(混洗)第一个索引,其中dset[0][0]包含输入图像张量,dset[0][1]包含相应的标签或目标。