| title | emoji | colorFrom | colorTo | sdk | sdk_version | app_file | pinned | license |
|---|---|---|---|---|---|---|---|---|
Pytorch Resnet34 Bird Classification |
👀 |
yellow |
gray |
gradio |
3.16.1 |
app.py |
false |
gpl-3.0 |
The project is an implementation of the ResNet34 model as per the Microsoft Research Paper. The model is build using PyTorch and is trained on the Birds Classification Dataset from Kaggle.
All the code for training the model and exporting to ONNX format is present in the notebook folder or you can use this Kaggle Notebook for training the model. It took ~1.5 hours to train the model on the complete dataset using a P100 GPU. The app.py file contains the code for deploying the model using Gradio.
You can try out the model on Hugging Face Spaces
