代码之家 › 专栏 › 技术社区 › tidy

tf.summary.image文件似乎不适用于估计器预测

tensorflow-estimator tensorflow python

2

tidy · 技术社区 · 6 年前

我想可视化我的输入图像使用tf.估计量什么时候预测,但似乎tf.summary.image文件不保存图像。但它对训练有用。

...
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 ?

1 回复 | 直到 6 年前

1

2

xdurch0 6 年前

我假设您需要将摘要(直方图/图像) 之前 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])
...