Skip to content

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

Notifications You must be signed in to change notification settings

peterbacalso/african_wildlife_classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

About

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

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages