Question

目前我试图更熟悉TF Estimator API。我正在使用此blog entry中的示例工作/学习。

现在我遇到的问题是我无法保存简单的图像摘要（tf.summary.image）。

我在 experiment_fn 中添加了 SummarySaverHook ，并将钩子添加到Experiment构造函数的 eval_hooks 参数中：

def experiment_fn(run_config, params):
   ...
   summary_hook = tf.train.SummarySaverHook(
      100,
      output_dir='/summeries',
      summary_op=tf.summary.merge_all()
   )
   ...
   experiment = tf.contrib.learn.Experiment(
      estimator=estimator,  # Estimator
      train_input_fn=train_input_fn,  # First-class function
      eval_input_fn=eval_input_fn,  # First-class function
      train_steps=params.train_steps,  # Minibatch steps
      min_eval_frequency=params.min_eval_frequency,  # Eval frequency
      train_monitors=[train_input_hook],  # Hooks for training
      eval_hooks=[eval_input_hook, summary_hook],  # Hooks for evaluation
      eval_steps=None  # Use evaluation feeder until its empty
   )

我不清楚的是我必须添加 tf.summary.image（）功能，以便图像显示在张量板的图像标签中。

提前致谢。

Answer 1

你不应该添加一个钩子。只需在tf.summary.image / model_fn中的任意位置添加input_fn来电，估算工具就会自动为所有已创建的摘要添加摘要挂钩。

使用Estimator API保存tf.summary.image

1 个答案: