- minimum python 3.7
- minimum pip 19.0
- Clone the repo
- Create virtual env:
python -m venv venv
- Install requirements:
pip install -r requirements
- Enter the
src
folder - Train the model
python train_digit_classifier.py --model ./output/digit_classifier.h5
Note: The model is already trained, you can play with parameters to get other results