导出推理图和冻结图之间的区别?

时间:2018-02-20 23:29:18

标签: tensorflow

对于张量流模型,有两个类似的脚本,export_inference_graph和freeze_graph。在训练我们自己的模型的情况下,有什么区别以及应该使用哪一个?谢谢

1 个答案:

答案 0 :(得分:4)

根据我的理解,export_inference_graph是对象检测研究项目的自定义脚本。(您将拥有对象检测模型,配置文件,类似的东西......)
freeze_graph用于冻结通用模型:删除不必要的节点,将变量转换为常量等 因此,在您的情况下,您需要使用freeze_graph作为一般模型。