chrismachado / lenet-cnn

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

LeNet Convolutional Neural Network

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

Usage

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 

Author

• Christiano Machado ([e-mail] | github.com/chrismachado)

References

This tutorial was made by Adrian Rosebrock.