Skip to content

An API for identifying cougars v.s. bobcats v.s. other USA cat species

Notifications You must be signed in to change notification settings

simonw/cougar-or-not

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

cougar-or-not

My first attempt at a machine learning API, using a pre-calculated model trained using iNaturalist data.

The resulting API is used by the @critter_vision Twitter bot, the source code for which can be found at https://github.com/natbat/CritterVision

The model is usa-inaturalist-cats.pth - an 83MB file.

The notebook inaturalist-cats.ipynb shows how I trained the model, using fastai.

cougar.py is a very tiny Starlette API server which simply accepts file image uploads and runs them against the pre-calculated model.

It also accepts a URL to an image, e.g. https://cougar-or-not.now.sh/classify-url?url=https://upload.wikimedia.org/wikipedia/commons/9/9a/Oregon_Cougar_ODFW.JPG

The Dockerfile means the entire thing can be deployed to Zeit Now or any other container hosting service.

Examples

Cougar: https://cougar-or-not.now.sh/classify-url?url=https://upload.wikimedia.org/wikipedia/commons/9/9a/Oregon_Cougar_ODFW.JPG

Bobcat: https://cougar-or-not.now.sh/classify-url?url=https://upload.wikimedia.org/wikipedia/commons/thumb/d/dc/Bobcat2.jpg/1200px-Bobcat2.jpg

About

An API for identifying cougars v.s. bobcats v.s. other USA cat species

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Sponsor this project

 

Packages

No packages published

Languages