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

nissan/cougar-or-not

 
 

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 99.8%
  • Other 0.2%