在Android上跑TensorFlow之接入TensorFlow

本文介绍了如何在Android应用中接入TensorFlow运行框架。通过在TensorFlow源码中编译so包和jar包,然后利用TensorFlowInferenceInterface类进行模型加载、输入、运行和输出的操作。具体步骤包括编译so包、编译jar包和调用相关接口进行推理流程。详细实现可参考官方示例代码。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

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


三、调用步骤:

调用的接口,都封装在TensorFlowInferenceInterface类中,在 tensorflow\contrib\android\java\org

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值