RuntimeError: Found 0 images in subfolders of: ./data

在使用torchvision.datasets.ImageFolder(path, transform)进行加载图片时报错。出错代码如下。

dataset = torchvision.datasets.ImageFolder('../cocotrain/', transform=data_transform)
data_loader = torch.utils.data.DataLoader(dataset, batch_size=batch_size, shuffle=True)

报的错误是：

RuntimeError: Found 0 images in subfolders of: ../cocotrain/
Supported image extensions are: .jpg,.JPG,.jpeg,.JPEG,.png,.PNG,.ppm,.PPM,.bmp,.BMP.

解决方案：

在我的代码里，我把cocotrain文件夹下的图片文件都存放在了文件夹名为1的文件里。之前我的目录结构是

/cocotrain/
--1.jpg
--2.jpg
...

修改之后我的目录结构是

/cocotrain/1/
--1.jpg
--2.jpg
...

参考资料：

RuntimeError: Found 0 images in subfolders of: ./data