Skip to content

An image classifier that predicts members of the genus Panthera family. Trained using FastAI v2 API and deployed on StreamLit.

Notifications You must be signed in to change notification settings

dnaveenr/big_cat_classifier

Repository files navigation

The Big Cats Classifier

Big Cats

The term "big cat" is typically used to refer to any of the five living members of the genus Panthera, namely the tiger, lion, jaguar, leopard, and snow leopard, as well as the non-pantherine cheetah and cougar.

Upload an image of one of the Big Cat and determine which category the big cat belongs to. The model is trained using FastAI with backbone of Resnet18 architecture. Image data is retrieved using the Bing Image Search API.

Side Note

I have always had a problem in recognizing big cats, so I thought of building a classifier using FastAI as part of my learning from the FastAI book.

Categories the Classifier can predict :

  1. Cheetah
  2. Cougar
  3. Jaguar
  4. Leopard
  5. Lion
  6. Snow leopard
  7. Tiger

Results

The model gets an accuracy of 91% on the validation set, the model does get confused between the leopard and jaguar categories.

Live Demo

The model is deployed using StreamLit Sharing. You can try it out here :
Streamlit

References

  1. FastAI Book.
  2. Streamlit deployment is inspired from here.

About

An image classifier that predicts members of the genus Panthera family. Trained using FastAI v2 API and deployed on StreamLit.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published