PyTorch ImageFolder：文件夹获取哪个索引？答案

【问题标题】：PyTorch ImageFolder: Which Index Does a Folder Get?PyTorch ImageFolder：文件夹获取哪个索引？
【发布时间】：2021-09-03 22:45:57
【问题描述】：

我想创建一个混淆矩阵，如 here 所示，尽管我使用的数据集与 FashinMNIST 不同。具体来说，我希望轴包含诸如“香蕉”、“橙子”、“黄瓜”等名称。我有一个存储香蕉等图像的文件夹，我使用PyTorch's ImageFolder class 加载该文件夹。现在，PyTorch 自动将数字 0 分配给我的班级（我总共有 10 个班级），但我不确定标签“香蕉”是否总是得到相同的数字 0。

在使用 ImageFolder 类时，如何找出哪个索引属于哪个标签（子文件夹名称）？

【问题讨论】：

标签： machine-learning pytorch dataset

【解决方案1】：

类ImageFolder 具有属性class_to_idx。检查docs。

x = torchvision.datasets.ImageFolder(root=path, transform=transform)
print(x.class_to_idx)

【讨论】：