This is a set of classification models (VGG, Resnet) trained on Animals-10 - a dataset of ~28k images of animals of 10 different categories.
It's still in progress as I'm still currently training the models. Weights for Resnet-101 and VGG-16 will be out soon.
Resnet-50 achieves the 80.7% val acc
This project primarily uses Pytorch and FastAPI.
VGG Architecture can be found and explored here Resnet Architecture can be found and explored here