Handwritten Digit Classification Android app

Overview

This project comprises of a handwritten character image (MNIST) classifier that can run on any android device. The app stores a set of images (0-9) that we can cycle through and classify in order. It uses a pre-trained model to perform inference on the device. This idea can be applied to any images, both by using the camera and by pulling from the Web. We're using preloaded images so we can run the app in a simulator (no need for the device since it doesn't require a camera).

I used the popular MNIST dataset for this task. The significant focus is on deploying the trained model in Android framework, rather than training a classification model. If you need to understand the task of classifying handwritten digits, this is a good starting point.

Results

Here are a few output results, which depict the UI of the app along with the classification results.

The model predicts the output label along with the confidence score for the prediction.

Setting up Environment

Installation Steps

If you already have Android Studio, skip to Step 5!

Download Android studio https://developer.android.com/studio/index.html

Download Android SDK

$ wget https://dl.google.com/android/android-sdk_r24.4.1-linux.tgz
$ tar xvzf android-sdk_r24.4.1-linux.tgz -C ~/tensorflow

Download SDK Build Tools

$ cd ~/tensorflow/android-sdk-linux
$ tools/android update sdk --no-ui

Download Android NDK

$ wget https://dl.google.com/android/repository/android-ndk-r12b-linux-x86_64.zip
$ unzip android-ndk-r12b-linux-x86_64.zip -d ~/tensorflow

Set correct SDK and NDK versions in your workspace file

Train Model in Python on your desktop or a server tensorflow_model/mnist_convnet_keras.py
Download the latest Tensorflow Android release package (AAR) AAR (Android Archive) files are a convenient way to distribute packages– mainly libraries– for use with Android Studio and Gradle. Adding one of these files to your app requires creating some metadata files and updating your app’s gradle build files. Fortunately, Android Studio can handle all of this for us. http://docs.onemobilesdk.aol.com/android-ad-sdk/adding-aar-files.html
Compile the code and run it on a simulator (or device if you have one)!

Exporting a pre-trained model

Train your model
Keep an in memory copy of eveything your model learned (like biases and weights)

Example: _w = sess.eval(w),

where w was learned from training.

Rewrite your model changing the variables for constants with value = in memory copy of learned variables.

Example: w_save = tf.constant(_w)

Also make sure to put names in the input and output of the model, this will be needed for the model later.

Example:
x = tf.placeholder(tf.float32, [None, 1000], name='input')
y = tf.nn.softmax(tf.matmul(x, w_save) + b_save), name='output')

Export your model with:

tf.train.write_graph(<graph>, <path for the exported model>, <name of the model>.pb, as_text=False)

Credits

Thanks to Siraj Raval (@sirajology) for teaching an amazing course on Deep Learning at Udacity, covering this material and more.