如果所有图像都在一个文件夹中，如何使用pytorch读取.csv文件以进行“培训类别”？

Question

在this tutorial中，他们使用以下代码读取数据集：

data_dir = 'data/hymenoptera_data'
image_datasets = {x: datasets.ImageFolder(os.path.join(data_dir, x),
                                          data_transforms[x])
                  for x in ['train', 'val']}
dataloaders = {x: torch.utils.data.DataLoader(image_datasets[x],
                                              batch_size=4,
                                              shuffle=True,
                                              num_workers=4)
               for x in ['train', 'val']}

在该代码中，png数据集位于不同的文件夹中。

每个文件夹的名称表示该文件夹中png文件的类别。

---

我处于以下情况：

• 所有png文件都在一个文件夹中；
• 每个png文件的类别都记录在train.csv文件中。

如何使用上述功能datasets.ImageFolder实现代码？