文章目录
写一段我将tensorflow移植到Android上的历程,希望后者能够避免我的坑,能更多的发展这个东西。
1,按照以前的经验,需要在Ubuntu上用tensorflow源码安装,再用bazel编译出
写一段我将tensorflow移植到Android上的历程,希望后者能够避免我的坑,能更多的发展这个东西。
过程中参考了其他一些博客,外加自己的修改,填坑,由于可能比较长,剩下的之后再补充吧。
采用的策略是tensorflow moblie
但是谷歌开发会议上提出了Tensorflow Lite相信不久这个将会是主流
本篇主要针对移植的方面说明,对Android和tf的细节并未做详细说明
本篇不恰当之处还请指正,欢迎评论区指出不足之处
大体的路线是:
- 试图在Ubuntu上使用Bazel编译jar文件和so文件(包含Git下载源码,源码安装)
- 在Ubuntu上安装Android Studio
- 安装NDK
- 放弃编译,直接下载jar和so文件(太难编译了)
- tensorflow保存pb文件
- Ubuntu上成功
- windows上成功
我的环境
- Windows10
- Ubuntu 16虚拟机
- tensorflow版本:win(1.8.0-gpu) Ubuntu(1.8.0-cpu)
- Python版本 :均为3.6
- Android Studio3.x
- 真机测试环境:小米5X
一,先在Ubuntu虚拟机上操作
1,按照以前的经验,需要在Ubuntu上用tensorflow源码安装,再用bazel编译出
- libandroid_tensorflow_inference_java.jar(以下简称jar文件)
- libtensorflow_inference.so(以下简称so文件)
这两个文件是tensorflow移植到Android的关键,外加Tensorflow训练保存的pb文件
3.model.pb(以下简称pb文件)
2,安装Android Studio、NDK和Bazel
AS:直接在官网安装即可