用估计API保存tf.summary.image

Question

目前，我试图对更加熟悉一些。我正在学习这个博客条目中的例子。

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

我在SummarySaverHook中添加了一个experiment_fn，并将钩子添加到实验构造函数的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()函数，以便将图像显示在tensorboard的image选项卡中。

提前谢谢。

Answer 1

你不应该再加个钩子。只需将tf.summary.image调用添加到model_fn/input_fn中的任何地方，估计器就会自动为创建的所有摘要添加一个摘要挂钩。