Awesome
African Wildlife
Minimal machine learning API, inspired by fastai MOOC and cougar-or-not api
app/main.py
is a small API built with fastapi. It takes an image url or file and runs it through a pretrained model for classification between buffalo, elephant, rhino, or zebra.
export.pkl
is a resnet34 model, pretrained with the fastai framework using kaggle African Wildlife dataset. See african_wildlife.ipynb
for training details.