我使用以下方法将一系列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()
我得到以下图片:
这是我的期望。然后我用:
加载到Tensorboardtf.summary.image('input', x, 1)
但图片显示为:
图片还可以,但颜色都搞砸了。有什么想法会发生什么?
答案 0 :(得分:2)
没关系,导致问题的是CV2.imread。 CV2读取BGR而不是RGB,因此错误加载的图像和Tensorflow正在显示它们。