在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 目录下,如下图所示:


本文介绍了如何将训练好的TensorFlow模型迁移到Android应用中,通过在Android Studio中配置TensorFlow Lite,将pb模型放入assets目录,添加必要的库文件,并更新build.gradle配置,包括设置multiDexEnabled,abiFilters,以及引入libandroid_tensorflow_inference_java.jar。
最低0.47元/天 解锁文章

被折叠的 条评论
为什么被折叠?



