Image classification powered by a neural network implemented with Keras and TensorFlow.js
Try it out yourself at https://anna-charlotte.github.io/digit_recognizer/frontend/index.html
Train model:
- Clone repository:
git clone https://github.com/anna-charlotte/digit_recognizer.git - Install requirements:
poetry install - Activate virtual environment:
poetry shell - Create the data directory:
mkdir data - Download the data from https://www.kaggle.com/datasets/oddrationale/mnist-in-csv?resource=download and save it to the data directory.
- Create additional training data:
python src/create_new_data.py - Train model:
python src/digit_recognizer.py