Overview

In this project, I have developed an application that, given an image of a dog, it will identify an estimate of the canine’s breed. If supplied an image of a human, the code will identify the resembling dog breed.

Installation

1. Clone the repository and navigate to the downloaded folder.

git clone https://github.com/xquyvu/dog-breed-detector.git
cd dog-breed-detector

2. Get the trained models

• The model can be downloaded here
• Save the trained model as ./model_transfer.pt

3. Install requirements

   pip

   pip install requirements.txt
   

   conda

   conda env create -f deep-cv-pytorch.yml
   

Usage

Simply run main.py and provide the path to your image.

If you haven't had VGG16 before, the first run will download VGG16 to your machine, which results in longer run time.

Example: python main.py --img_path ./images/Curly-coated_retriever_03896.jpg

To do

Document how the model was trained, results, and future work