/mnist_digit_recognizer

Draw any number on canvas to get it segmented and classified

Primary LanguagePythonOtherNOASSERTION

MNIST Classifier

Requires python v3

Instructions

  • Create a python venv and activate
$ python -m venv testEnvironment
$ cd testEnvironment
$ ./Scripts/activate
  • Clone this repository into the venv
$ git clone https://github.com/faztp12/mnist_digit_recognizer.git
  • Install this on the virtual env
$ pip install -e ./mnist_digit_recognizer
  • Train the models, optionally
$ python -m mnist_digit_recognizer
$ python -m mnist_digit_recognizer.conv # for conv network
  • Start the GUI
$ python -m mnist_digit_recognizer.gui