python - tf.summary.image 似乎不适用于估计器预测
问题描述
我想在预测时使用 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
吗?
解决方案
我假设您需要在调用之前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])
...
推荐阅读
- node.js - 带默认参数的模块导出
- python - 如何在python中使用分组键将OrderedDict列表转换为嵌套json
- macos - 在 NSViewControllerRepresentable 中使用 SwiftUI Coordinator 和 NSTableView?
- java - 使用 JPA 将枚举值存储到单独的 Oracle 属性中
- php - 我无法用我的 phpmailer 发送 pdf 附件
- javascript - 有没有办法同时托管 react 和 vanilla javascipt 网页?
- android - 有条件地在 xml 布局中包含/添加视图?
- c++ - 没有规则为 googletest 制定目标
- dll - Windows 上的 Python setuptools 帮助。未识别 dll 依赖项
- html - 微软边缘多元素副本