yerim2 / tflite-mnist-android

MNIST with TensorFlow Lite on Android

MNIST with TensorFlow Lite on Android

This project demonstrates how to use TensorFlow Lite on Android for handwritten digits classification from MNIST.

Prebuilt APK can be downloaded from here.

How to build from scratch

Environment

• Python 3.6, TensorFlow 1.13.1
• Linux or macOS for tflite model conversion.

Step 1. Train and convert the model to TensorFlow Lite FlatBuffer

Run all the code cells in model.ipynb.

• If you are running Jupyter Notebook locally, a mnist.tflite file will be saved to the project directory.
• If you are running the notebook in Google Colab, a mnist.tflite file will be downloaded.

Step 2. Build Android app

Copy the mnist.tflite generated in Step 1 to /android/app/src/main/assets, then build and run the app. A prebuilt APK can be downloaded from here.

The Classifer reads the mnist.tflite from assets directory and loads it into an Interpreter for inference. The Interpreter provides an interface between TensorFlow Lite model and Java code, which is included in the following library.

implementation 'org.tensorflow:tensorflow-lite:1.13.1'

If you are building your own app, remember to add the following code to build.gradle to prevent compression for model files.

aaptOptions {
    noCompress "tflite"
    noCompress "lite"
}

Credits

• The basic model architecture comes from tensorflow-mnist-tutorial.
• The official TensorFlow Lite examples.
• The FingerPaint from Android API demo.