This repository is a simple flask app which runs a custom ResNet neural network trained on the CIFAR100 dataset. Given an input image, this application returns one of CIFAR's 100 image classifications.
>>> docker build -t image-class:latest .
>>> docker run -p 8080:8080 image-class:latest
>>> python3 -m venv .venv
>>> source .venv/bin/activate
>>> pip install -r requirements.txt
>>> python image_classification_flask_app/main.py
The training and model designed was largely built from this excellent jovian.ai tutorial by user @Hurly119. App design is from a simple flask template by @roytuts.