Skip to content

Tensorflow Image classifier that can predict the breed of a dog from it photo. Trained on image dataset of 5 different breed of dogs (rottweiler, bulldog, pug, german shepherds, labrador). Interestingly the classifier was able to predict the breed of the dogs even from images of their toys.

Notifications You must be signed in to change notification settings

AthulDilip/Tensorflow-Dog-Breed-Classifier

Repository files navigation

Tensorflow Dog Breed Classifier

For all those dog lovers like me out there, here is a Tensorflow image classifier that can preict the breed of a dog from its photo. This model is trained to detect 5 different breeds of dog.

  1. Rottweiler
  2. Bulldog
  3. Pug
  4. German Shepherds
  5. Labrador

Intrestingly enough, the model was able to classify the dog breed even from the picture of its toy. Feel free to fork and play around with the classifier.

How to run the basic model

For an image located at <path-to-image-file>, run :

python classifier.py --image_path <path-to-image-file>

About

Tensorflow Image classifier that can predict the breed of a dog from it photo. Trained on image dataset of 5 different breed of dogs (rottweiler, bulldog, pug, german shepherds, labrador). Interestingly the classifier was able to predict the breed of the dogs even from images of their toys.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages