TensorFlow for Android PB模型的优化

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