Skip to content

aditi-dsi/DogBreedClassifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pre-trained Image Classifier to identify dog breeds

Implemented pre-trained CNN model-architectures to distinguish Dogs from everything else in this world :), followed by identifying their correct breeds.
In addition to this, I also implemented and compared the results from 3 great architectures, namely, AlexNet, ResNet, & VGG and found out which one works the best in most of the cases (including edge cases).


(I tested the model with my own images as well, you can do that too !!)

How to take model inference for your own images?

  1. Move inside 'uploaded_images' directory

  2. Delete(or keep) previous files and upload new images.
    (Note: Image should be Sqaure in shape, i.e. their length and breadth should be similar px.)

  3. Open the terminal and run the below script :

    
    $ sh run_models_batch_uploaded.sh
    
    

Suggestions to get the most out of it:

Try experimenting with variety of images such as :

  1. animals other than dog,
  2. any random object,
  3. or a same picture with different angle would work too !

Comparison results among AlexNet, ResNet, VGG:

image

Results clearly show that VGG shines out.. 🎉🌟

For a more detailed version of the results, check out the respective files for each model in the following format : <model_name>_pet-images.txt & <model_name>_uploaded-images.txt





That's all about this project: Created with ❤️ by Aditi
✨Feel free to suggest any improvements, provide feedback for my work !!

About

Implemented Pre-trained Image Classifier for Detecting Dogs and their Breeds.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published