1

I want to create a confusion matrix as it is shown here, though I use a different dataset than FashinMNIST. Concretely, I want the axes to contain names such as "banana", "orange", "cucumber", etc. I have a folder where images of bananas, etc. are stored, which I load with PyTorch's ImageFolder class. Now, PyTorch automatically assings the numbers 0 to 9 to my classes (I have 10 classes in total), but I'm not sure whether the label "banana" always gets the same number 0.

How do I find out, when using the ImageFolder class, which index belongs to which label (subfolder name)?

desertnaut
  • 52,940
  • 19
  • 125
  • 157
Hermi
  • 159
  • 2
  • 10

1 Answers1

2

Class ImageFolder has attribute class_to_idx. Check the docs.

x = torchvision.datasets.ImageFolder(root=path, transform=transform)
print(x.class_to_idx)
Dharman
  • 26,923
  • 21
  • 73
  • 125
Trong Van
  • 354
  • 1
  • 12