Pytorch：无法使用 ImageFolder 加载图像
Pytorch: Can’t load images using 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）。
[torch.FloatTensor of size 3x64x64]
文件夹中有超过 10,000 张图片。怎么存不下所有数据？
你应该试试这个：
print len(dset)
表示数据集的大小，也就是图像文件的数量。
dset[0] 表示数据集的（打乱）第一个索引，其中 dset[0][0] 包含输入图像张量，dset[0][1] 包含相应的标签或目标。
我正在尝试使用“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 张图片。怎么存不下所有数据？
你应该试试这个：
print len(dset)
表示数据集的大小，也就是图像文件的数量。
dset[0] 表示数据集的（打乱）第一个索引，其中 dset[0][0] 包含输入图像张量，dset[0][1] 包含相应的标签或目标。