tensorflow:估计器导出器输出与pbtxt

时间:2018-09-16 11:26:04

标签: python unity3d tensorflow

我正在努力导出训练有素的模型,以使其能够统一使用。我阅读的每本指南都具有相同的说明-

  1. 保存图表
  2. 冻结图形(pbtxt + chckpt)
  3. 优化冻结图

例如this guide

另一方面,我使用估算器构建了模型,因此认为导出模型的更自然的方法是导出器:

    export = tf.estimator.FinalExporter('exporter',
    serving_input_receiver_fn=tf.estimator.export.build_raw_serving_input_receiver_fn(
        {'img': tf.placeholder(tf.float32, [None, 3])}))


    eval_spec = tf.estimator.EvalSpec(
        input_fn=_define_input_fn(),
        exporters=export)

上面的代码生成两种类型的输出:

saved_model.pb
variables/

我想知道如何使用这些输出,我发现的所有指南都提到这些输出仅用于在服务器中进行部署。

这些文件是否需要“冻结”?优化?我的导出输出与常规图形和chckpt文件有什么区别?

谢谢! 尼桑。

0 个答案:

没有答案