Dog's breed Classifier and Similar Dogs search engine. An Flask app to classify Dog's type and Find similar dogs. Instructions Run Classification:
- Classification/Dogs.ipynb
- Then save the model- Dogs.h5
run feature_extractor.py
python3 server.py
This codebase tries to present in the simplest and most compact way a one of the major Transfer Learning techniques, which have emerged over the past years. This project contains two main things :- Classification with the help of Transfer Learning and Features extraction for Image similarity. Flask is used for serving these models/ Features.
For Yolo you can download from - https://pjreddie.com/darknet/yolo/
To get the weights file of Dogs breed classifier mail me at [email protected] (Or just train it. Colab will be very useful for this task :) )
