The dog breed dataset consists of 133 different breeds
-
Download data and unzip:
$ wget https://s3-us-west-1.amazonaws.com/udacity-aind/dog-project/dogImages.zip
$ unzip dogImages.zip
- Clone project
$ git clone https://github.com/DucLeTrong/dog_breed_classification
$ cd dog_breed_classification
- Install requirements
$ pip3 install -r requirements.txt
- Train model
$ python3 train.py
$ python3 test.py
$ python3 infer.py --img_path='2.jpg' --model_path='model.pt'
- If a dog is detected in the image, return the predicted breed.
- If a human is detected in the image, return the resembling dog breed.
- If neither is detected in the image, provide output that indicates an error.