想把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模型。