PyTorch ImageFolder: 处理单个目录下的图片数据集

如果您在使用 PyTorch 的 ImageFolder 类时，发现您的目录直接包含很多图片，而不是按类别分组的文件夹，您可以尝试使用 ImageFolder 的 from_folder 方法来创建数据集。

以下是一个示例代码：

import torchvision.datasets as datasets

data_dir = 'path/to/your/directory'
transform = ... # 定义你的图像转换操作

dataset = datasets.ImageFolder.from_folder(data_dir, transform=transform)

这样，dataset 将包含您目录下所有的图片，并且每个图片都将被归类为一个类别。