/african_wildlife_classifier

API for identifying buffalos, elephants, rhinos, or zebras in African wildlife

Primary LanguageJupyter Notebook

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.