TensorFlowLite:将TensorFlow训练好的模型迁移到Android APP上

本文介绍了如何将训练好的TensorFlow模型迁移到Android应用中,通过在Android Studio中配置TensorFlow Lite,将pb模型放入assets目录,添加必要的库文件,并更新build.gradle配置,包括设置multiDexEnabled,abiFilters,以及引入libandroid_tensorflow_inference_java.jar。
摘要由CSDN通过智能技术生成

在Android Studio中配置TensorFlowLite
  把训练好的 TensorFlow的 pb模型  移植到Android Studio上,TensorFlow Lite官方的例子中已经给我们展示了,我们其实只需要两个文件: libandroid_tensorflow_inference_java.jar、libtensorflow_inference.so。

        这两个文件我已经放到github上了,大家可以自行下载使用,下载地址:

                                    libandroid_tensorflow_inference_java.jar、libtensorflow_inference.so。

  然后首先把训练好的pb模型放到Android项目中app/src/main/assets下,若不存在assets目录,则自己新建一个。如图所示

          其次,把刚刚下载的 libandroid_tensorflow_inference_java.jar 文件放到 app/libs 目下,把libtensorflow_inference.so 放到 app/libs/armeabi-v7a 目录下,如下图所示: 

然后在app/build.gradle里进行如下配置: 
(i)   在

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值