TensorFlow for Android(安卓)PB模型的优化
16lz
2021-01-26
想把TensorFlow迁移到Android平台上,由于手机计算能力有限,因此需要对PB模型进行优化。
使用windows系统下自带的优化py文件,路径:
D:\Anaconda3\Lib\site-packages\tensorflow\python\tools\optimize_for_inference.py
在D:\Anaconda3\Lib\site-packages\tensorflow下打开命令行,将训练完成的retrained_graph.pb放到这里,然后输入:
python -m tensorflow.python.tools.optimize_for_inference \ --input=retrained_graph.pb \ --output=optimized_graph.pb \ --input_names="Mul" \ --output_names="final_result"
可以看到已经正确生成相应的optimized_graph.pb优化后的文件。
再用这个生成后的pb进行测试,能够正确使用。
三 放入Android中测试
替换app\src\main\assets\model中对应的PB文件与txt文件,然后进行测试,成功加载PB模型。
更多相关文章
- 在Android中让Preference的宽度占满整个屏幕的宽度
- Android应用性能优化
- Android布局优化几个重要标签的使用include、merge、ViewStub、r
- [Android] ramdisk.img的生成及解压
- Android(安卓)网络框架Volley JAR包生成
- 如何在Android(安卓)NDK中调用第三方库文件(.so)
- Android代码实现WiFi连接
- android中生成keystore
- NDK HOWTO