Skip to content

Latest commit

 

History

History
34 lines (23 loc) · 1.33 KB

README.md

File metadata and controls

34 lines (23 loc) · 1.33 KB

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.