Skip to content

A CNN built from scratch and another one from transfer learning to recognize dog breed for dogs and humans

License

Notifications You must be signed in to change notification settings

masoodkhanpatel/Dog-Breed-Recognition

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Dog-Breed-Recognition

In this project, I have bult a pipeline to process real-world, user supplied images. A pipeline that can be used within a web or mobile app. Given an image of a dog, our algorithm will idenity an estimate of the canine's breed. If supplied an image of a human, the code will identify the resembling dog breed.

  • The datasets that we are using is of dogs and humans to distinguish between them.

  • We used OpenCV's implementation of Haar feature-based cascade classifiers to detect human faces in images. OpenCV provides many pre-trained face detectors, stored as XML files on github. We have downloaded one of these detectors and stored it in the haarcascades directory.

  • We used a pre-trained model to detect dogs in images. Specifically VGG-16 model, along with weights that have been trained on ImageNet, a very large, very popular dataset used for image classification and other vision tasks. ImageNet contains over 10 million URLs, each linking to an image containing an object from one of 1000 categories.

  • We also used our own custom CNN. Sample CNN...

The following links may be of great interest for learning:

  1. Convolutional Neural Network With PyTorch
  2. How To Improve Deep Learning Performance
  3. A Beginner’s Guide to Recurrent Networks and LSTMs
  4. Introduction to Recurrent Neural Networks in Pytorch
  5. Understanding LSTM Networks

Further Resources and Imrpovements

About

A CNN built from scratch and another one from transfer learning to recognize dog breed for dogs and humans

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published