python - Torchvision ImageFolder“找不到任何类文件夹”

Question

下面的代码

plastic_train_image_folder = torchvision.datasets.ImageFolder(plastic_dir, transform=transforms)

引发以下错误：

在 /Users/username/Documents/Jupyter/archive/垃圾分类/垃圾分类/塑料中找不到任何类文件夹。

然而，那里有文件。下面的代码打印482.

list_plastic = os.listdir(plastic_dir) 
number_files_plastic = len(list_plastic)
print(number_files_plastic)

为什么会发生此错误？

标签： pythonpytorchtorchvision

Answer 1

正如您在文档中看到的那样，ImageFolder该类期望图像位于目录中，每个感兴趣的类都有一个：

以这种方式排列图像的通用数据加载器：

root/dog/xxx.png
root/dog/xxy.png
root/dog/xxz.png

root/cat/123.png
root/cat/nsdf3.png
root/cat/asd932_.png

您的图像可能位于根目录中，这不是它所期望的方式，因此错误。