/stanford-dogs

PyTorch code for the stanford dogs dataset

Primary LanguagePythonMIT LicenseMIT

Stanford Dogs

I wrote a model to train on the stanford dogs dataset, and also tried transfer learning with a resnet50 model after that. The code for the transfer learning is almost the same as from this tutorial, with just very few changes.

I achieved an accuracy of 82.11% with the resnet50 model.