[转载]Android Studio编译TensorFlow的Demo

原文地址： Android Studio编译TensorFlow的Demo 作者： 奋斗的点心要专注

TensorFlow是Google的机器学习开源框架，可以使用Git获取源代码。

https://github.com/tensorflow/tensorflow.git

 

其中AndroidDemo的代码位于Examples下， 具体的编译规则 可以到官方Markdown页面查看： https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/android

 

以下是简要说明：

 

1. 编译Native代码

Linux/Mac系统：

下载Bazel对源代码进行编译： https://github.com/bazelbuild/bazel/releases

此外，NDK是必备的： https://developer.android.com/ndk/index.html，也可以使用Android Studio的Settings -> Appearance & Behavior -> System Settings -> Android SDK，切换到SDK Tools的Tab下，选中NDK进行安装。

接下来编译代码：

CPU=armeabi-v7a

bazel build //tensorflow/examples/android:tensorflow_native_libs --crosstool_top=//external:android/crosstool --cpu=$CPU --host_crosstool_top=@bazel_tools//tools/cpp:toolchain

NATIVE_FOLDER=tensorflow/examples/android/libs/$CPU

mkdir -p $NATIVE_FOLDER

cp bazel-bin/tensorflow/examples/android/libtensorflow_demo.so $NATIVE_FOLDER

 

Windows：

Bazel还不能在Windows平台下编译Android Native，可以使用预编译好的so：

https://ci.tensorflow.org/view/Nightly/job/nightly-android/lastSuccessfulBuild/artifact/

将编译好的so放入libs目录下。

注释掉根目录build.gradle下的所有task。

 

 

2. 下载训练数据

从 https://storage.googleapis.com/download.tensorflow.org/models/inception5h.zip下载数据后，解压到Demo项目的Assets目录下。

 

3. 用Android Studio导入Demo项目，编译运行