如何使用 ImageFolder 加载 ImageNet 数据集

作者建议您使用 torchvision.datasets.ImageFolder 类加载 ImageNet 数据集，而不是原代码中的方法。

您需要删除代码中的第 156 行到 164 行，并用以下代码替换它们：

root = os.path.join(config.DATA.DATA_PATH, prefix)
dataset = datasets.ImageFolder(root, transform=transform)

这将使用 ImageFolder 类加载数据，该类接受一个目录路径和一个变换函数。 ImageFolder 类将自动将目录中的图像文件加载到数据集，并使用变换函数对图像进行预处理。

作者还提到他们将在稍后更新代码。