README.md

metadata

title: Pytorch Resnet34 Bird Classification
emoji: 👀
colorFrom: yellow
colorTo: gray
sdk: gradio
sdk_version: 3.16.1
app_file: app.py
pinned: false
license: gpl-3.0

Pytorch Resnet34 Bird Classification

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.

🚀 Getting Started

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.

🤗 Demo

You can try out the model on Hugging Face Spaces

🖥️ Sample Interface