python - 将图像摘要输出到 Tensorboard

Question

我正在尝试使用 this code 可视化卷积层的滤波器，但无法将图像写入我的摘要文件。我有没有问题输出的标量，并且我尝试修改代码以添加图像，如下所示

summary = tf.Summary()
summary.value.add(tag='Perf/Reward', simple_value=float(mean_reward))
summary.value.add(tag='Perf/Length', simple_value=float(mean_length))
with tf.variable_scope(self.name + "/conv1", reuse=True):
    weights = tf.get_variable("weights")
    grid = put_kernels_on_grid(weights)
    image = tf.summary.image('conv1/weights', grid, max_outputs=1)
    summary.value.add(tag='conv1/weights', image=image)
self.summary_writer.add_summary(summary, episode_count)

仅使用标量，效果很好，但添加图像会出现错误

TypeError: Parameter to MergeFrom() must be instance of same class: expected Image got Tensor. for field Value.image

我还尝试通过将代码更改为

直接添加图像摘要

summary = tf.Summary()
summary.value.add(tag='Perf/Reward', simple_value=float(mean_reward))
summary.value.add(tag='Perf/Length', simple_value=float(mean_length))
with tf.variable_scope(self.name + "/conv1", reuse=True):
    weights = tf.get_variable("weights")
    grid = put_kernels_on_grid(weights)
    image = tf.summary.image('conv1/weights', grid, max_outputs=1)
    self.summary_writer.add_summary(image, episode_count)
self.summary_writer.add_summary(summary, episode_count)

但出现错误

AttributeError: 'Tensor' object has no attribute 'value'

将图像输出到摘要文件的正确方法是什么？

Answer 1

put_kernels_on_grid 返回一个张量；我认为作者所说的“图像”只是意味着您可以将其打印出来以查看内核的样子。尝试使用tf.summary.tensor_summary。

哎呀，抱歉。我已经能够使用以下命令将张量保存为图像:

import tensorflow as tf
import numpy as np

batch_xs = np.ones((100, 100, 1)) * 200
init = tf.constant(batch_xs, dtype=tf.uint8)
grid = tf.get_variable('var_name', dtype=tf.uint8, initializer=init)

encoded_image = tf.image.encode_jpeg(grid)

fwrite = tf.write_file("junk.jpeg", encoded_image)
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    result = sess.run(fwrite)

但是tf.image.encode_jpeg仍然返回一个带有DataType字符串的张量，因此tf.summary.image不会接受它。您正在使用的代码早于 TensorFlow 1.0，因此它肯定不会按编写的方式工作。

希望这能有所帮助。