The libraries required for the successful execution of this code are mentioned in requirements.txt. In order to install all the libraries:
pip install -r requirements.txt
Download the dog images from here and extract into Data/dogImages
folder.
Download the human images from here and extract into Data/humanImages
folder. Rename the lfw folder as humanImages.
Download the bottleneck features for the ResNet50 model from here and put them into `Data/bottleneck_features/' folder.
In this project, I have built a dog breed classifier that classifies dog images into their respective breeds. If the image is of a human, the classifier predicts the most resembling dog breed.
The jupyter notebook contains an in depth analysis of the dataset and the project. It iterates between different solutions before coming to the final solution.
Finally, I have created a Flask web app that allows a user to upload an image and predicts the breed.
To run the web app, first run train.py
in order to train the model and save the model for future prediction.
Finally, to run the web app simply type flask run
into the terminal.
The data was provided by Udacity. Otherwise, feel free to use the code here as you would like!