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

Question

我正在使用以下方法将一系列 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()

我得到了下图：

这是我所期望的。然后我加载到 Tensorboard：

tf.summary.image('input', x, 1)

但图像显示是：

图像还可以，但颜色都乱七八糟。知道会发生什么吗？

Answer 1

没关系，是 CV2.imread 导致了问题。 CV2 读取 BGR 而不是 RGB，因此图像加载不正确，而 Tensorflow 却按应有的方式显示它们。

Tensorboard tf.summary.image strange image colors