PyTorch ImageFolder: Which Index Does a Folder Get?
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 &quot;banana&quot;, &quot;orange&quot;, &quot;cucumber&quot;, 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 &quot;banana&quot; always gets the same number 0.
How do I find out, when using the ImageFolder class, which index belongs to which label (subfolder name)?

Class ImageFolder has attribute class_to_idx. Check the docs.
x = torchvision.datasets.ImageFolder(root=path, transform=transform)
print(x.class_to_idx)

