tensorflow - 在 TensorBoard 中可视化层的激活

Question

我有一个层，layer3它是类型：

Tensor("vgg_16/conv3/conv3_3/Relu:0", shape=(1, 500, 700, 120), dtype=float32, device=/device:GPU:0)

我想可视化这一层的激活。我该如何处理layer3呢？我必须添加什么才能将tf.summary.histogram()其可视化？

Answer 1

首先，在所需层上定义您的摘要：

tf.summary.histogram("layer3_hist_summary", layer3)

接下来定义将用于将摘要写入磁盘的摘要编写器：

LOGDIR = 'path/to/logsdir' # define your required summary output folder
summary_file_writer = tf.summary.FileWriter(logdir=LOGDIR)

假设您可能有多个摘要，请将它们合并到一个操作中：

summary_op = tf.summary.merge_all()

现在，在您的训练循环中，编写摘要结果：

for i in range(NUM_ITR):
    _, summary_res = sess.run([train_op, summary_op])
    summary_file_writer.add_summary(summary_res, global_step=i)

要查看这些摘要，请使用logsdir=LOGDIR.