Tensorboard tf.summary.image奇怪的图像颜色

时间:2018-01-26 14:23:03

标签: python image-processing tensorflow tensorboard

我使用以下方法将一系列128x128 uint8彩色图像加载到Tensorflow:

x = tf.placeholder(tf.float32, shape=[None, img_size,img_size,num_channels], name='x-input')

我首先检查我加载的图像是否正常所以我使用Matplotlib检查它(注意Matplotlib使用BGR颜色,所以我在显示之前修复它):

plt.imshow(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
plt.show()

我得到以下图片:

enter image description here

这是我的期望。然后我用:

加载到Tensorboard
tf.summary.image('input', x, 1)

但图片显示为:

enter image description here

图片还可以,但颜色都搞砸了。有什么想法会发生什么?

1 个答案:

答案 0 :(得分:2)

没关系,导致问题的是CV2.imread。 CV2读取BGR而不是RGB,因此错误加载的图像和Tensorflow正在显示它们。