AI MNIST Canvas

This is a simple python application that uses a neural network to predict the number drawn on a canvas. Also, it shows the probabilities of each number and the features maps of the neural network.

How to use

Install the requirements:

pip install -r requirements.txt

Train the neural network:

python -m mnist_cnn.train

Run the application:

python ./main.py

Demonstration

demo.mp4