PIL.Image和cv2的图片读取转换

这里读取一个RGB的图片

img = Image.open(imgpath)
print(img,img.size())
img.show()

print的结果如下,可以看到是没有第三波段的,只显示出了H和W


(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来展示,但是你会发现现实的图片颜色十分诡异。

PIL.Image和cv2的图片读取转换_第1张图片

原因在于cv2不仅要求接受的是H,W,C的矩阵,而且要求通道必须是BGR的顺序,而Image读取后的图片是RGB的。因此还需要变换一下RGB的通道。这里有两种方法:

  • 直接变换channel维度的顺序
img = img[:, :, [2, 1, 0]]   #原本是RGB->BGR
  • 采用cv2自带的函数
img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)

 

你可能感兴趣的:(python,python,opencv)