python - "RuntimeError: Found 0 files in subfolders of ".. Pytorch 中的子文件夹错误

Question

我目前基于 Window 10、Jupyter Notebook、Pytorch 1.0、Python 3.6.x。

首先，我使用以下代码确认文件的正确路径:print(os.listdir('./Dataset/images/'))。

我可以检查这条路径是否正确。

但我遇到了Error :

RuntimeError: Found 0 files in subfolders of: ./Dataset/images/ Supported extensions are: .jpg,.jpeg,.png,.ppm,.bmp,.pgm,.tif"

怎么了？您能提出解决方案吗？

我试过 ./dataset/1/images 喜欢这个方法。但结果是一样的....

img_dir = './Dataset/images/'
img_data = torchvision.datasets.ImageFolder(os.path.join(img_dir), transforms.Compose([
            transforms.Scale(256),
            transforms.RandomResizedCrop(224),
            transforms.RandomHorizontalFlip(),
            transforms.ToTensor(),
            ]))
img_batch = data.DataLoader(img_data, batch_size=batch_size,
                               shuffle = True, drop_last=True)

Answer 1

我在使用celebA时遇到了同样的问题，包括200,000张图片。正如我们所看到的，有很多图像。但是在小样本情况下(我试了20张图片)，我检查了一下，不会报错，说明我们可以成功读取图片。但是当数量增长时，我们应该使用其他方法。

我根据这个网站解决了这个问题。感谢 QimingChen Github solution

简单地说，在原始文件夹中添加另一个名为 1 (/train/--->train/1/) 的文件夹将使我们的程序能够运行，无需更改路径。那是因为当面对大数据集时，图像应该被分类到不同类别的子文件夹中。

Github 上的原始答案:

假设我要使用 ImageFolder("/train/") 读取文件夹 train 中的 jpg 文件。文件结构是/火车/-- 1.jpg-- 2.jpg-- 3.jpg

我未能加载它们，导致错误:RuntimeError:在以下子文件夹中找到 0 个图像:./data支持的图片扩展名有:.jpg,.JPG,.jpeg,.JPEG,.png,.PNG,.ppm,.PPM,.bmp,.BMP

我阅读了上面的解决方案并尝试了数十次。当我将结构更改为/train/1/

-- 1.jpg-- 2.jpg-- 3.jpg

但是读入的代码仍然是 -- ImageFolder("/train/")，它工作正常。

程序似乎倾向于递归读取文件，这在某些情况下很方便。

希望这会有所帮助!