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