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

扩展Keras TensborBoard回调以可视化模型预测,keras,image-segmentation,tensorboard,Keras,Image Segmentation,Tensorboard,在训练深层语义分段模型时,在训练期间可视化验证集上的预测样本通常很方便。现在我只是将一些预测保存到培训服务器上的磁盘上。我希望将此任务迁移到TensorBoard。简单地说,我不想想象每个时代的一组预测(比如5) 我知道在纯TensorFlow中有一种简单的方法可以做到这一点,比如tf.summary.image(…),但我看不到任何简单的方法可以将其合并到Keras TensorBoard回调中 如果您能提供任何指导,我们将不胜感激。法比奥·佩雷斯提供了一个答案,这个答案应该完全符合您在这里的

在训练深层语义分段模型时,在训练期间可视化验证集上的预测样本通常很方便。现在我只是将一些预测保存到培训服务器上的磁盘上。我希望将此任务迁移到TensorBoard。简单地说,我不想想象每个时代的一组预测(比如5)

我知道在纯TensorFlow中有一种简单的方法可以做到这一点,比如
tf.summary.image(…)
,但我看不到任何简单的方法可以将其合并到Keras TensorBoard回调中


如果您能提供任何指导,我们将不胜感激。

法比奥·佩雷斯提供了一个答案,这个答案应该完全符合您在这里的要求