image - すべての画像が1つのフォルダにある場合、pytorchで「トレーニングカテゴリ」の.csvファイルを読み取る方法は?

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 ファイルは 1 つのフォルダーにあります。
• 各 png ファイルのカテゴリは train.csv ファイルに記録されます。

上記の関数を使用してコードを実装するにはどうすればよいdatasets.ImageFolderですか?