我可以通过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上查看这些图像？

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

Answer 1

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