前言
使用cv2打开图片默认是BGR模式,而使用PIL显示图片时是RGB模式,如果不做转换,图片会出现色彩上的问题,例如:
cv_img = cv2.imread('2022-05-06.png')
PIL.Image.fromarray(np.uint8(cv_img)).show()
BGR转RGB
使用cv2.imread打开图片会得到一个三维numpy数组,这个数组最后一维有3个数字分别代表这个图片在这个像素点上的BGR(默认模式)的数值。所以只要将这三个数字交换位置即可转换模式。
下面是将图片numpy数组BGR转RGB的函数代码:
def BGR_to_RGB(cvimg):
pilimg = cvimg.copy()
pilimg[:, :, 0] = cvimg[:, :, 2]
pilimg[:, :, 2] = cvimg[:, :, 0]
return pilimg
在使用PIL显示图片前调用这个函数即可使颜色恢复正常:
cv_img = cv2.imread('2022-05-06.png')
pil_img = BGR_to_RGB(cv_img)
PIL.Image.fromarray(np.uint8(pil_img)).show()
执行结果:
总结
BGR转RGB其实就是将代表B和R的数字交换位置。将代表B,G,R的数字交换位置就能得到其它颜色模式下的图片数组。