Pytorch中torchvision.datasets.ImageFolder的Found 0 files in subfolders错误

报错如下:

RuntimeError: Found 0 files in subfolders of: /home/A/0
Supported extensions are: .jpg,.jpeg,.png,.ppm,.bmp,.pgm,.tif,.tiff,webp

产生原因:路径问题

举例说明: /home/A/0/ 目录下存在若干图片,/home/A/ 目录下只有 0 这一个文件夹。那么ImageFolder的root应该选择哪一个imFolder呢???选择标红一个,产生bug的原因就是选择了 标蓝的一个。

imgFolder=/home/A

imgFolder=/home/A/0

ImageFolder(root=imgFolder, .......)

 

你可能感兴趣的:(Pytorch中torchvision.datasets.ImageFolder的Found 0 files in subfolders错误)