0

在本教程中，他们使用以下代码来读取数据集：

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？

4

0 回答 0