如何使用tf.summary.image在Keras回调中仅保存一些自定义图像？

Question

我正在使用一个自定义回调，该回调创建一些matplotlib图形并将其泵送到tf.summary.image。

我还使用了如下的张量板回调：

tb_callback = tf.keras.callbacks.TensorBoard(
    log_dir=tb_output_dir,
    histogram_freq=10,
    write_graph=True,
    write_images=True, # how to write only the plot images?
    embeddings_freq=0,
    embeddings_layer_names=None,
    embeddings_metadata=None,
    embeddings_data=None,
    update_freq='epoch', # batch epoch or int
    profile_batch=2
    )

有什么方法只能推送matplotlib映像吗？看来它全有还是全无。