本文是TensorFlow官网教程:将 TensorFlow 深度学习模型,移植到Android手机,进行物体识别、行人检测等具体实现过程(Android demo官方主页:https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/android)。
一、下载Tensorflow源码
将源码克隆一份到本地,或直接打开链接下载到本地。
注:--recurse-submodules参数用来防止后面出现protobuf错误。
二、搭建环境
1、安装构建工具Bazel
本文采用了Bazel custom APT安装。参考网站https://docs.bazel.build/versions/master/install-ubuntu.html进行。
1)安装JDK 8:
2)添加APT库
3)安装和更新Bazel
如果只是更新,则运行:
2、安装SDK
2)方式二:通过Android Studio安装(方式一28.0.0若报错可用方式二)。
3、安装NDK
4、修改WORKSPACE
tensorflow根目录下打开WORKSPACE,找到如下一段代码,对应配置好自身的SDK版本以及位置,还有NDK版本及位置,并去掉注释:
三、编译apk并安装
进入tensorflow目录下,运行:
bazel build -c opt //tensorflow/examples/android:tensorflow_demo
编译成功后会在目录下面生成文件
安装apk到手机,用数据线连上手机,执行:
adb install -r bazel-bin/tensorflow/examples/android/tensorflow_demo.apk
或直接将apk文件放置到手机文件点击安装。
四、运行程序
TF Detect
TF Stylize
领取 专属20元代金券
Get大咖技术交流圈