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

Question

我已将Tensorboard与Keras设置为回调,如下所示:

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上查看这些图像？

我tf.summary.image在https://github.com/tensorflow/tensorboard上找到了这个功能 但我不明白这与Keras的关系.任何帮助,将不胜感激.

Answer 1

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