Как прочитать файл .csv для "категории обучения" с pytorch, если все изображения находятся в одной папке?

Question

В этом руководстве они используют следующий код для чтения наборов данных:

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?