如何在张量板中绘制历史图像

Question

我想在tensorboard中的一个滑块下显示图像。以下是我的代码，

import io
import matplotlib.pyplot as plt
import seaborn as sns
import tensorflow as tf
import numpy as np


def gen_plot(): 
    """Create a pyplot plot and save to buffer."""
    uniform_data = np.random.rand(10, 12)
    plt.figure()
    sns.heatmap(uniform_data)   
    plt.title("test")
    buf = io.BytesIO()
    plt.savefig(buf, format='png')
    buf.seek(0)
    return buf

with tf.Session() as sess:
    for step in range(10):
        plot_buf = gen_plot()
        image = tf.image.decode_png(plot_buf.getvalue(), channels=4)
        image = tf.expand_dims(image, 0)
        image = tf.summary.image("plot", image)
        summary_op = tf.summary.merge_all()
        writer = tf.summary.FileWriter('./logs')
        writer.add_summary(summary_op.eval(), step)
writer.close()

但是，这就是我得到的Tensorboard images。在tensorboard中创建多个图像而不是一个。并且一个滑块下的所有图像都是相同的。

我是TF的新手。对不起，如果这个问题有点愚蠢。但是，如何使用步骤滑块更改我的代码以查看历史图像？