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

Question

在训练深度语义分割模型时，在训练期间可视化验证集上的预测样本通常很方便。现在我只是将一些预测保存到我的训练服务器上的磁盘上。我希望将此任务迁移到 TensorBoard。简而言之，我不想在每个时期可视化一组预测（比如 5 个）。

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

任何指导将不胜感激。

标签： kerasimage-segmentationtensorboard

Answer 1

Fabio Perez 提供了一个答案，它应该完全符合您在此处寻找的内容如何使用 Keras 在 TensorBoard 中显示自定义图像？