为Android优化TF模型

时间:2018-09-18 12:33:09

标签: tensorflow tensorflow-datasets tensorflow-lite

当我尝试优化模型时,我遇到了优化模型以在Android设备上运行的问题;

  

raise KeyError(“找不到以下输入节点:%s \ n”%   not_found)KeyError:“未找到以下输入节点:   设置(['Mul'])\ n“

允许我的代码进行优化;

python C:\tensorflow\tensorflow\python\tools\optimize_for_inference.py --input=c:\tf_files\retrained_graph.pb --output=c:\tf_files\optimized_graph.pb --input_names="Mul" --output_names="final_result"

我曾经尝试将“ Mul”更改为“ Input”,但收到了同样的错误。

enter image description here

0 个答案:

没有答案