机器学习:如何在安卓上集成TensorFlow
我们都知道,谷歌有一个开源库叫做TensorFlow,可被用在安卓系统中实现机器学习。换言之,TensorFlow是谷歌为机器智能提供的一个开源软件库。
我在网络上搜寻了很久,都没有找到在安卓上搭建TensorFlow的简单的方法或实例。仔细查阅许多资料以后,我终于可以搭建它了。于是,我决定把我搭建的过程写出来,这样其他人就不必再浪费时间了。
这篇文章是写给那些熟悉机器学习并且知道怎样为机器学习搭建模型的人的(在这个示例中我会使用一个预训练模型)。近期,我会写一系列关于机器学习的文章,这样每个人都能够学到如何为机器学习搭建模型。
从搭建安卓上的机器学习模型过程讲起
我们需要知道的几个要点:
-
TensorFlow的核心是用C++编写的;
-
为了在安卓上搭建TensorFlow,我们需要用JNI(Java本地接口)来调用C++函数,比如说loadModel,getPredictions,等等;
-
我们会用到.so(shared object,即共享对象)文件,它是C++编译文件;还会用到jar文件,它由能够调用本地C++的Java API组成。之后,我们就可以调用Java API轻松地把事情做好;
-
所以我们需要jar(Java API)和一个.so(C++编译)文件;
-
我们必须要有一个预训练模型文件和一个用于分类的标签文件。
我们会做以下的目标检测:
编译jar和.so文件
注意:--recurse-submodules对于提取子模块(pull submodules)很重要。
在这里(https://developer.android.com/ndk/downloads/older_releases.html#ndk-12b-downloads)下载NDK。
下载安卓SDK,或者,我们也可以从Android Studio SDK提供路径。
安装Bazel(https://bazel.build/versions/master/docs/install.html)。Bazel是TensorFlow的主要编译系统。
现在,编辑工作空间(WORKSPACE),我们可以在早先克隆的TesnsorFlow根路径中找到工作空间(WORKSPACE)文件。
我们的SDK和NDK路径就跟下面一样:
然后编译生成.so文件:
将armeabi-v7a换成我们所需要的目标架构。
库会被放置在:
编译Java副本:
我们可以在这里找到JAR文件:
现在我们有了jar和.so文件。你也可以从下面的工程中直接提取使用我已经建立好的.so文件和jar。
我已经在这里(https://github.com/MindorksOpenSource/AndroidTensorFlowMachineLearningExample)创建了一个完整可运行的示例应用。
但是,我们需要预训练模型和标签文件。
在这个例子中,我们会使用Google预训练模型,它实现了在一张给定的照片上做目标检测。
解压缩zip文件,我们就会得到imagenet_comp_graph_label.strings.txt(目标标签)以及tensorflow_inception_graph.pb(预训练模型)。
现在,在Android Studio上创建安卓示例工程吧。
将imagenet_comp_graph_label.strings.txt(目标标签)以及tensorflow_inception_graph.pb放进assets文件夹。
将libandroid_tensorflow_inference_java.jar放进lib文件夹,单击右键,添加库。
在主目录新建一个jniLibs文件夹并且将libtensorflow_inference.so放到jniLibs/armeabi-v7a文件夹中。
现在,我们就可以调用TensorFlow Java API了。
TensorFlow Java API通过TensorFlowInferenceInterface类开放了所有需要的方法。
现在,我们可以用模型路径调用TensorFlow Java API并且加载它了。
然后,我们可以输入一张图片来获取预测结果。
如果想要体会完整的流程,克隆这个项目(https://github.com/MindorksOpenSource/AndroidTensorFlowMachineLearningExample),搭建并运行它吧。
如果你在搭建这个项目的过程中有任何问题的话,联系我,我会非常乐意帮助你。
Happy Coding:)
读者问答
Q:我很疑惑要怎么连接到‘so’库?也没有任何一行像’System.loadLibray’的代码?
A:System.loadLibrary已经写在TensorFlow jar中了。
Q:所以这是一个C++应用还是Java应用啊?你用的是什么语言?
A:这是一个用Java语言编写的安卓应用,它通过Java本地接口(JNI: Java Native Interface)调用C++做预测(机器学习)。
原文发布时间为:2017-04-27
本文来自云栖社区合作伙伴“大数据文摘”,了解相关信息可以关注“BigDataDigest”微信公众号
相关文章
- 大数据挖掘机器学习人工智能的维恩图战争
- 关于程序员面试的那些事
- 如何快速掌握一门新技术/语言/框架
- 大数据发展到现在,面临的主要挑战是什么?
- 如何选择数据平台的建设方案
- 数据科学家的离职潮背后,有怎样的扎心真相?
- “大数据杀熟”未必都错,且看它背后的定价思维
- 如何面试你的面试官
- 程序员世界常见的6个问题
- 泼冷水:为什么如此多的数据科学家正在抛弃他们的工作?
- 详解:大数据分析的学习之路
- 千万别让九种失误毁了你的面试表现
- 如何选择正确的数据集成工具
- 资源控制在大数据和云计算平台中的应用
- 加班能体现编程的热情吗?
- 数据科学简介:分数据、结构和数据科学管道(一)
- 为什么编程那么难?是我脑子不行吗?
- 我多希望我学编程时,有人教我这些事!
- 高端大气的“数据科学家”,竟然成了离职率最高的职位之一
- “你中有我,我中有你”,人工智能想和大数据谈一场恋爱!