This is a learning project that aims to create a model using CNN LeNet. The model will classify the famous MNIST data set.

Cloning this repo using

git clone https://github.com/chrismachado/lenet-cnn

Install python dependencies with pip

pip install -r requirements.txt 

Then, if you just want to train and test the model, just type

python lenet_mnist.py 

Or, you can run pre-trained model in output folder by using the following command

python lenet_mnist.py --load-model 1 --weights output/lenet_weights.hdf5 

For more information about the script, you can use help command

python lenet_mnist.py -h 

• Christiano Machado ([email protected] | github.com/chrismachado)

This tutorial was made by Adrian Rosebrock.