/big-cat-app

Simple fastai/starlette web app for big cat recognition

Primary LanguageJupyter Notebook

Which one of the wild cats is in the image

Simple web application for image classification. Running on https://wild-cat-app.herokuapp.com (and runs out of memory almost all the time). Input an image, and model tries to predict which of the species of the extended big cat family. Possible labels are

  • Caracal
  • Cheetah
  • Clouded leopard
  • Cougar
  • Jaguar
  • Jungle cat
  • Leopard
  • Lion
  • Lynx
  • Serval
  • Snow leopard
  • Sunda clouded leopard
  • Tiger

Network used here is retrained resnet34, refitted with self-collected dataset of big cats (approximately 3000 images, about 150-370 samples for each species). Network training is done with fast.ai.

Adapted from https://github.com/pankymathur/google-app-engine