tf.summary.image, кажется, не работает для прогнозирования оценки

Question

Я хочу визуализировать мое входное изображение, используя tf.estimator при прогнозировании, но кажется, что tf.summary.image не сохраняет изображение. Но это работает для обучения.

Это мой код в model_fn:

...
summary_hook = tf.train.SummarySaverHook(
        save_secs=2,
        output_dir='summary',
        scaffold=tf.train.Scaffold(summary_op=tf.summary.merge_all()))
        #summary_op=tf.summary.merge_all())
tf.summary.histogram("logit",logits)
tf.summary.image('feat', feat)
if mode == tf.estimator.ModeKeys.PREDICT:
    return tf.estimator.EstimatorSpec(mode, predictions=preds, prediction_hooks=[summary_hook])
...

и это мой код предсказания:

config = tf.estimator.RunConfig(save_summary_steps=0)
estimator = tf.estimator.Estimator(model_fn=model_fn, model_dir='logs', config=config)
preds = estimator.predict(input_fn=eval_input_fn)

Что-то не так для использования tf.train.SummarySaverHook?

Answer 1

Я бы предположил, что вам нужно поместить сводные операции (гистограмма / изображение) перед вызовом merge_all, чтобы merge_all действительно что-то слилось.

...
tf.summary.histogram("logit",logits)
tf.summary.image('feat', feat)
summary_hook = tf.train.SummarySaverHook(
    save_secs=2,
    output_dir='summary',
    scaffold=tf.train.Scaffold(summary_op=tf.summary.merge_all()))
    #summary_op=tf.summary.merge_all())
if mode == tf.estimator.ModeKeys.PREDICT:
    return tf.estimator.EstimatorSpec(mode, predictions=preds, prediction_hooks=[summary_hook])
...