参考链接:
- tensorflow 物体检测模型相关资料 https://github.com/tensorflow/models/tree/master/research/object_detection
- java api接口 https://tensorflow.google.cn/lite/apis#java
Running on mobile with TensorFlow Lite
以物体检测模型为例,生成一个APK文件,可以调用TensorFlow Lite模型,进而实现手机端能够运行物体检测模型。
整体分为以下几个步骤:
步骤一、训练一个TensorFlow 模型(或者从下载他人训练好的模型)
步骤二、生成.pb文件
步骤三、将.pb文件转化为.tflite文件
步骤四、使用Android Studio 导入TensorFlow安卓样例 ,替换成我们生成的tflite文件,重新编译APK
#配置依赖
https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/installation.md
#下载样例模型