我可以通过Keras将图像添加到Tensorboard吗？-Java 学习之路

我已将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的关系 . 任何帮助，将不胜感激 .

1 回答

0

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

回复于 2024-05-17T17:35:45+08:00

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

1 回答

相关问题