Deploying a TensorFlow MNIST model on Arm NN

Overview Load and parse the MNIST test set Import graph Optimize and load onto compute device Run graph on device Deploy an application using Arm NN Next steps
Single page Download PDF

Load and parse the MNIST test set

The example code starts by loading and parsing the MNIST test set:

Code for loading a test image and its label

The loadMnistImage helper function is not covered here. In simple terms, it just parses the MNIST data files and returns an MnistImage struct for the requested image with a label and a 28*28=784 element array containing the data:

Code - Helper struct for loading MNIST data

Previous Next