作者 | 武维
审校 | 蔡芳芳
AI前线出品｜ ID：ai-front
前言
深度学习在图像处理、语音识别、自然语言处理领域的应用取得了巨大成功，但是它通常在功能强大的服务器端进行运算。如果智能手机通过网络远程连接服务器，也可以利用深度学习技术，但这样可能会很慢，而且只有在设备处于良好的网络连接环境下才行，这就需要把深度学习模型迁移到智能终端。
由于智能终端CPU和内存资源有限，为了提高运算性能和内存利用率，需要对服务器端的模型进行量化处理并支持低精度算法。TensorFlow版本增加了对Android、iOS和Raspberry Pi硬件平台的支持，允许它在这些设备上执行图像分类等操作。这样就可以创建在智能手机上工作并且不需要云端每时每刻都支持的机器学习模型，带来了新的APP。
本文主要基于看花识名APP应用，讲解TensorFlow模型如何应用于Android系统；在服务器端训练TensorFlow模型，并把模型文件迁移到智能终端；TensorFlow Android开发环境构建以及应用开发API。
看花识名APP
使用AlexNet模型、Flowers数据以及Android平台构建了“看花识名”APP。TensorFlow模型对五种类型的花数据进行训练。如下图所示：
模型训练及模型文件
本章采用AlexNet模型对Flowers数据进行训练。AlexNet在2012取得了ImageNet最好成绩，top 5准确率达到80.2%。这对于传统的机器学习分类算法而言，已经相当出色。模型结构如下：
本文采用TensorFlow官方Slim（https://github.com/tensorflow/models/tree/master/slim）AlexNet模型进行训练。
量化操作的主要思想是在模型的Inference阶段采用等价的8位整数操作代替32位的浮点数操作，替换的操作包括：卷积操作、矩阵相乘、激活函数、池化操作等。量化节点的输入、输出为浮点数，但是内部运算会通过量化计算转换为8位整数（范围为0到255）的运算，浮点数和8位量化整数的对应关系示例如下图所示：
量化Relu操作的基本思想如下图所示:
TensorFlow Android应用开发环境构建
在Android系统上使用TensorFlow模型做Inference依赖于两个文件libtensorflow_inference.so和libandroid_tensorflow_inference_java.jar。这两个文件可以通过下载TensorFlow源代码后，采用bazel编译出来，如下所示：
1、下载TensorFlow源代码git clone --recurse-submoduleshttps://github.com/tensorflow/tensorflow.git
TensorFlow提供了Android开发的示例框架，下面基于AlexNet模型的看花识名APP做一些相应源码的修改，并编译生成Android的安装包：
3、编译生成安装包
4、拷贝tensorflow_demo.apk到手机上，并执行安装，太阳花识别效果如下图所示：
TensorFlow移动端应用开发API
在Android系统中执行TensorFlow Inference操作，需要调用libandroid_tensorflow_inference_java.jar中的JNI接口，主要接口如下：
3、执行模型推理； outputNames为TensorFlow Inference模型中要运算Tensor的名称，本APP中为分类的Logist值。
总结
本文基于看花识名APP，讲解了TensorFlow在Android智能终端中的应用技术。首先回顾了AlexNet模型结构，基于AlexNet的slim模型对Flowers数据进行训练；对训练后的CheckPoint数据，进行Freeze和Quantized处理，生成智能终端要用的Inference模型。然后介绍了TensorFlow Android应用开发环境的构建，编译生成TensorFlow在Android上的动态链接库以及java开发包；文章最后介绍了Inference API的使用方式。
参考文献
4、深度学习利器：TensorFlow系统架构与高性能程序设计
作者简介
武维（微信：3381209@qq.com）：博士，系统架构师，主要从事大数据，深度学习，云计算等领域的研发工作。
－全文完－
人工智能已不再停留在大家的想象之中，各路大牛也都纷纷抓住这波风口，投入AI创业大潮。那么，2017年，到底都有哪些AI落地案例呢？机器学习、深度学习、NLP、图像识别等技术又该如何用来解决业务问题？
2018年1月11-14日，AICon全球人工智能技术大会上，一些大牛将首次分享AI在金融、电商、教育、外卖、搜索推荐、人脸识别、自动驾驶、语音交互等领域的最新落地案例，应该能学到不少东西。目前大会8折报名倒计时，更多精彩可点击阅读原文详细了解。
网友评论