我可以通过 Keras 将图像添加到 Tensorboard 吗？

Posted 2023-03-27

【中文标题】我可以通过 Keras 将图像添加到 Tensorboard 吗？

【英文标题】：Can I add images to Tensorboard through Keras?

【发布时间】：2018-07-01 10:58:28

【问题描述】：

我已经使用 Keras 作为回调设置了 Tensorboard，如下所示：

callback_tb = keras.callbacks.TensorBoard(log_dir=tb_dir, histogram_freq=2,write_graph=True,write_images=True)
callbacks_list = [callback_save,callack_view, callback_tb]
model.fit(x_train,y_train,
                  batch_size=batch_size,
                  epochs=epochs,
                  callbacks=callbacks_list,
                  verbose=1,
                  validation_data=(x_test,y_test),
                  shuffle='batch')

这很好用，我可以在 Tensorboard 上看到损失和准确度图。 我正在生成模型预测并将其保存在另一个文件中，但我想知道是否可以使用 Keras 在 Tensorboard 上查看这些图像？

我在https://github.com/tensorflow/tensorboard 上找到了tf.summary.image 函数 但我不明白这与 Keras 有什么关系。 任何帮助将不胜感激。

【问题讨论】：

您要添加哪些图片？

【参考方案1】：

我通过基于 TensorBoard 回调创建自己的 Keras 回调解决了我的问题，我可以在其中使用 tf.summary.image 功能

【讨论】：

嗨！我会对您的解决方案非常感兴趣。我尝试使用 tf.summary.image 修改来自 keras 的 Tensorboard 回调以显示预测输出，但无法使其工作......

老实说，将图像保存为 matplotlib 子图时我的运气会更好。在 Tensorboard 上显示图像非常笨拙且缓慢

@RomaneG 对于 saul，如果您能写出解决方案，我将不胜感激。我和 RomaneG 有同样的问题。而对于 RomaneG，你最终解决了这个问题吗？

@FallenApart 此问题已在此处的另一个线程中解决：***.com/questions/43784921/…