Pet Breed Classification Using CNNs

less than 1 minute read


Saw a cute dog or a beautiful cat and wonder what breed it is? This CNN model can help you.

Oxford cats and dogs dataset includes images of cats and dogs and annotations for their corresponding breeds.

I used the ResNet model in PyTorch, pretrained on the ImageNet dataset, and fintuned it to train a pet breed classifier. The web app is deployed here using Streamlit.

Check out the repository on GitHub or test the deployed web app and find out the breed for a pet you liked.

Please let me know if you see any mistakes or have any questions/suggestions. You could also follow me on Twitter and LinkedIn.

Thanks for reading!

Leave a Comment