扩展Keras TensborBoard回调以可视化模型预测

时间:2018-06-29 05:15:08

标签: keras image-segmentation tensorboard

在训练深度语义细分模型时,通常很容易在训练过程中将验证样本的可视化可视化。现在,我只是将一些预测保存到我的培训服务器上的磁盘上。我希望将此任务迁移到TensorBoard。简而言之,我不会在每个时期将一组预测(例如5)可视化。

我知道有一种简单的方法可以像tf.summary.image(..)那样在纯TensorFlow中进行操作,但是我看不到任何简单的方法可以将其合并到Keras TensorBoard回调中。

任何指导将不胜感激。

1 个答案:

答案 0 :(得分:0)

Fabio Perez提供了一个答案,该答案应该完全符合您在这里How to display custom images in TensorBoard using Keras?的要求