This project classifies images into two buckets: dog images and other images. It uses and compares the following models: vgg, resnet, alexnet.
python -m venv venv
source venv/bin/activate
pip install -r requirements.txt
python check_images.py --dir pet_images --arch vgg --dogfile dognames.txt
./run_models_batch.sh
./run_models_batch_uploaded.sh