How to get secific image folder into Datasets for Deeplearning Implementation

I am using cat breed image classification datasets from kaggle. datasets contains 40 classses of cat breed images into dataset folder, How could I choose specific 10 classes from 40 to load into Imagefolder.using index of folder label

Hey @lavkushgupt, you have to create Custom Dataset for that. You can look at this notebook to see how the author created a custom dataset. PyTorch-Tutorial (The Classification) | Kaggle

Thanks, this worked for me

1 Like