This Repo includes code (written in python ) for Stanford dogs data set classification . I have used PyTorch library for this purpose.
Top-1 Accuracy i have achieved is 76.66 % on Test set.
Top-5 Accuracy is 96.15% on Test set.
This project is done using Resnet18. I also have used Transfer learning i.e. I have used a pre-trained model of Resnet18 from torchvision models.
I have used GPU for training provided freely by Google Colab.(Thanks to Google on such a great help for researchers).