https://github.com/mikehquan19/animals_classifier
The model that can classify animals
https://github.com/mikehquan19/animals_classifier
computer-vision deep-learning fastapi image-classification pytorch
Last synced: 2 months ago
JSON representation
The model that can classify animals
- Host: GitHub
- URL: https://github.com/mikehquan19/animals_classifier
- Owner: mikehquan19
- Created: 2025-08-10T06:40:41.000Z (11 months ago)
- Default Branch: master
- Last Pushed: 2025-08-29T22:14:42.000Z (10 months ago)
- Last Synced: 2025-10-09T08:16:06.780Z (9 months ago)
- Topics: computer-vision, deep-learning, fastapi, image-classification, pytorch
- Language: Python
- Homepage:
- Size: 89.8 KB
- Stars: 0
- Watchers: 0
- Forks: 0
- Open Issues: 3
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Animals Classifier
This is a set of classification models (VGG, Resnet) trained on [Animals-10](https://www.kaggle.com/datasets/alessiocorrado99/animals10) - 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](https://arxiv.org/pdf/1409.1556)
Resnet Architecture can be found and explored [here](https://arxiv.org/pdf/1512.03385)