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

Question

下面的代码

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

抛出以下错误：

在 /Users/用户名/Documents/Jupyter/archive/Garbageclassification/Garbageclassification/ Plastic 中找不到任何类文件夹。

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

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

为什么会发生这个错误？

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

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