在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) 在