Skip to content

Latest commit

 

History

History
 
 

mnist

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 

MNIST - Handwritten Digit Recognition

Description

This model predicts handwritten digits using a convolutional neural network (CNN).

Model

Model Checksum Download (with sample test data) ONNX version Opset version
MNIST MD5 26 kB 1.0 1
MD5 26 kB 1.2 7
MD5 26 kB 1.3 8

Dataset

The model has been trained on the popular MNIST dataset.

Source

The model is trained in CNTK following the tutorial CNTK 103D: Convolutional Neural Network with MNIST. Note that the specific architecture used is the model with alternating convolution and max pooling layers (found under the "Solution" section at the end of the tutorial).

Demo

Run MNIST in browser - implemented by ONNX.js with MNIST version 1.2

Inference

We used CNTK as the framework to perform inference. A brief description of the inference process is provided below:

Input

shape (1x1x28x28)

Preprocessing

Output

shape (1x10)

Postprocessing

Route the model output through a softmax function to map the aggregated activations across the network to probabilities across the 10 classes.

Sample test data

Sets of sample input and output files are provided in

  • serialized protobuf TensorProtos (.pb), which are stored in the folders test_data_set_*/.

License

MIT