扩展 Keras TensborBoard 回调以可视化模型预测
Extending Keras TensborBoard callback to visualize model predictions
在训练深度语义分割模型时，在训练过程中可视化验证集上的预测样本通常很方便。现在我只是将一些预测保存到我的训练服务器上的磁盘上。我希望将此任务迁移到 TensorBoard。简而言之，我不想在每个时期可视化一组预测（比如 5 个）。
我知道在纯 TensorFlow 中有一种简单的方法可以做到这一点，例如 tf.summary.image(..)，但我没有看到任何简单的方法可以将其合并到 Keras TensorBoard 回调中。
任何指导将不胜感激。
Fabio Perez 提供了一个答案，应该完全符合您在此处寻找的内容
在训练深度语义分割模型时，在训练过程中可视化验证集上的预测样本通常很方便。现在我只是将一些预测保存到我的训练服务器上的磁盘上。我希望将此任务迁移到 TensorBoard。简而言之，我不想在每个时期可视化一组预测（比如 5 个）。
我知道在纯 TensorFlow 中有一种简单的方法可以做到这一点，例如 tf.summary.image(..)，但我没有看到任何简单的方法可以将其合并到 Keras TensorBoard 回调中。
任何指导将不胜感激。
Fabio Perez 提供了一个答案，应该完全符合您在此处寻找的内容