TensorFlow对移动设备的支持,挺好。
今天我们就来说说如何在Android应用中接入TensorFlow的运行框架。
在TensorFlow源码中,已经很好的包含了对支持Android运行的源码。
我们知道,要在Android中调用Native接口,需要so包和jar包(jni接口),
所以,我们先说说如何在TensorFlow源码中编译so包和jar包:
一、编译so包:
在TensorFlow源码根目录下运行:
bazel build -c opt //tensorflow/contrib/android:libtensorflow_inference.so \
--crosstool_top=//external:android/crosstool \
--host_crosstool_top=@bazel_tools//tools/cpp:toolchain \
--cpu=armeabi-v7a
编译好的so包,会存在:
bazel-bin/tensorflow/contrib/android/libtensorflow_inference.so
二、编译jar包:
在TensorFlow源码根目录下运行:
bazel build //tensorflow/contrib/android:android_tensorflow_inference_java
编译好的jar包,会存在:
bazel-bin/tensorflow/contrib/android/libandroid_tensorflow_inference_java.jar