/pre-trained-image-classifier-dog-breed

Pre-trained Image Classifier to Identify Dog Breeds implementation using Python

Primary LanguagePython

Pre-trained Image Classifier to Identify Dog Breeds

There is a repository to store my project for my nanodegree studies. It's a classifier implementation written in Python that identify dog breeds.

Samples Archs Scores

Resnet Alexnet VGG

Dependencies

Clone this repository.

$ git clone https://github.com/mmoraisa/pre-trained-image-classifier-dog-breed

Install the dependencies using PIP.

$ pip install -r requirements.txt

or

$ pip3 install -r requirements.txt

How it works

You can use it:

$ python check_images.py --dir pet_images/ --arch resnet  --dogfile dognames.txt > resnet_pet-images.txt

Command Line Arguments:

  1. Image Folder as --dir with default value 'pet_images'
  2. CNN Model Architecture as --arch with default value 'vgg'
  3. Text File with Dog Names as --dogfile with default value 'dognames.txt'

License

MIT

Free Software, Hell Yeah!