Pytorch中使用datasets.ImageFolder导入图片出现 Found 0 files in subfolders of: xxx 错误

问题

在执行这行代码的时候

train_data = datasets.ImageFolder('./data/image', transform = my_trans)

出现如下报错：

Traceback (most recent call last):
  File "D:/PyCharm Community Edition 2020.1.2/pytorch_projects/test1.py", line 16, in <module>
    train_data = datasets.ImageFolder('./data/image', transform = my_trans)
  File "D:\anaconda3\envs\pytorch_gpu\lib\site-packages\torchvision\datasets\folder.py", line 206, in __init__
    is_valid_file=is_valid_file)
  File "D:\anaconda3\envs\pytorch_gpu\lib\site-packages\torchvision\datasets\folder.py", line 98, in __init__
    "Supported extensions are: " + ",".join(extensions)))
RuntimeError: Found 0 files in subfolders of: ./data/image
Supported extensions are: .jpg,.jpeg,.png,.ppm,.bmp,.pgm,.tif,.tiff,.webp

报错的意思是 ./data/image 文件夹中没有满足如 .jpg，.png 等格式要求的文件，但实际上 image 文件夹中全是 .jpg 格式的图片，因此很困惑在网上搜索了半天原因。

原因

在看了这篇文章之后 Pytorch中torchvision.datasets.ImageFolder的Found 0 files in subfolders错误 觉得应该是路径有问题，应该改为图片所在路径的上一级路径

后来仔细分析报错原因， RuntimeError: Found 0 files in subfolders of: ./data/image， 这句话的意思是在 ./data/image 的子文件夹中找到0个文件，个人认为应该是在加载图片的时候系统去 image 下的子文件夹中找的？！希望懂的大佬可以留言！

解决

将 ‘./data/image’ 改为 ‘./data’ 就可以了，不过要注意最好 data 文件夹中只有一个文件夹，以便于系统搜索到要加载的图片。

参考：

1.解决PyTorch使用datasets.ImageFolder导入ImageNet数据集时出现的错误
2. Pytorch中torchvision.datasets.ImageFolder的Found 0 files in subfolders错误
3.Pytorch中torchvision.datasets.ImageFolder的Found 0 files in subfolders错误