python - 当数据格式为 "NCHW"时，如何在tensorboard中显示图像？

Question

在tensorflow中，使用函数tf.summary.image，我们可以在tensorboard中显示图像。

image( name, tensor, max_outputs=3, collections=None )  

变量tensor的数据格式为“NHWC”，即[batch_size, height, width,channels]。

但是当我使用“NCHW”([batch_size， channel ，高度，宽度])时，如何在张量板中显示图像？

Answer 1

图像摘要不处理多个内存布局，就像其他一些 tensorflow 函数一样。因此，目前，您唯一的选择就是转置图像。

由于tf.transpose是一个有点繁重的操作，您可以将转置限制为仅汇总的图像数量，例如summary_img_num:

summary_imgs = tf.transpose(imgs[:summary_img_num], perm=[0,3,1,2])
tf.summary.image('images', summary_imgs, max_outputs=summary_img_num)

如果您坚持使用tf.summary.image的默认max_outputs=3，

summary_imgs = tf.transpose(imgs[:3], perm=[0,3,1,2])
tf.summary.image('images', summary_imgs)