这里读取一个RGB的图片
img = Image.open(imgpath)
print(img,img.size())
img.show()
print的结果如下,可以看到是没有第三波段的,只显示出了H和W
<PIL.JpegImagePlugin.JpegImageFile image mode=RGB size=500x355 at 0x13E73B4ACC0>
(500, 355)
如果这时候想用cv2来进行展示,即cv2.imshow(img),就会出现错误。一方面,cv2接受的是一个H,W,C的矩阵(C,H,W的不行,pytorch中图片tensor都是C,H,W,转换成矩阵格式后同样无法直接用cv2显示)。此外,我们还需要对读入的图片进行相关的操作,这些操作也需要图片是矩阵的形式。因此这里将Image读取的图片转换为矩阵:
img = np.asanyarray(img)
print(img.shape)
此时输出的就是(500, 355,3)。此时可以用cv2来展示,但是你会发现现实的图片颜色十分诡异。
原因在于cv2不仅要求接受的是H,W,C的矩阵,而且要求通道必须是BGR的顺序,而Image读取后的图片是RGB的。因此还需要变换一下RGB的通道。这里有两种方法:
- 直接变换channel维度的顺序
img = img[:, :, [2, 1, 0]] #原本是RGB->BGR
- 采用cv2自带的函数
img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)