可视化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加载张量流。