Skip to content

nandahkrishna/CaffeMNIST

Repository files navigation

Caffe MNIST Example

This repository contains a LeNet Caffe Model trained on the MNIST dataset, based on the instructions available at https://caffe.berkeleyvision.org/gathered/examples/mnist.html.

Dataset

The MNIST test dataset is included in this repository, in the LMDB format. It was originally downloaded from http://yann.lecun.com/exdb/mnist and converted to the LMDB format using Caffe's convert_mnist_data tool.

Training

This model was trained using:

caffe train -solver=lenet_solver.prototxt

The train dataset is not included in this repository.

Testing

Run this to test the model:

caffe test -model=lenet_train_test.prototxt -weights=lenet_iter_10000.caffemodel

Performance

The model trained for 10000 iterations has a test accuracy of 98.6% and a softmax loss of 0.0444183.

License

The contents of this repository are released for unrestricted use.

About

Caffe MNIST model

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published