问当数据格式为"NCHW“时如何在tensorboard中显示图像？
EN

Stack Overflow用户

提问于 2017-07-09 11:40:37

回答 1查看 660关注 0票数 1

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

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

变量tensor的数据格式为"NHWC"，如[batch_size, height, width, channels]。

但是当我使用"NCHW“([batch_size, channels ,height, width])时，如何在tensorboard中显示图像？

回答已采纳

发布于 2017-07-09 14:27:45

图像摘要不像其他一些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)

票数 3

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/44992752

复制

相似问题

问当数据格式为"NCHW“时如何在tensorboard中显示图像？EN