Dog Breed Classifier

Hi everyone! Last month I was experimenting with fastai (v2) and made a CNN to predict a dog’s breed from a photo. While I started with the Stanford dogs dataset, I ended up making my own dataset from the internet with ~150 photos per dog, using the official AKC dog breeds list. When I was finished, I was so excited to find out that streamlit was starting out streamlit sharing! My friends and family have since uploaded photos of their dogs, and the model seems to work well on pure-bred dogs (which I’d expect) and sometimes hilariously on mixed-breed dogs.

Here’s the repo with my experiments, and you can access the streamlit app via https://willjobs.com/dogs (or the direct streamlit link).

Happy to hear any comments!

Best,
Will

3 Likes

Hey @willjobs!!

First, Welcome to the Streamlit community!!! :tada: :tada: :tada: :partying_face: :partying_face: :partying_face:

Love it when ML models do something hilarious! I tried your app and think its great!

Happy Streamlit-ing!!
Marisa

Thanks, Marisa! It’s so cool to see the results for people’s dogs :slightly_smiling_face:

1 Like