CODEWITHSUNDEEP
machine-learningpytorchdataset
Imahn
Imahn

Reputation: 534

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 "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)?

Upvotes: 1

Views: 2403

Answers (1)

Trong Van
Trong Van

Reputation: 382

Class ImageFolder has attribute class_to_idx. Check the docs.

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

Upvotes: 3

Related Questions