BGR转RGB(python)

BGR转RGB(python)

  • 前言
  • BGR转RGB
  • 总结


前言

使用cv2打开图片默认是BGR模式,而使用PIL显示图片时是RGB模式,如果不做转换,图片会出现色彩上的问题,例如:

cv_img = cv2.imread('2022-05-06.png')
PIL.Image.fromarray(np.uint8(cv_img)).show()

BGR转RGB(python)_第1张图片


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(python)_第2张图片


总结

BGR转RGB其实就是将代表B和R的数字交换位置。将代表B,G,R的数字交换位置就能得到其它颜色模式下的图片数组。

你可能感兴趣的:(python,python,计算机视觉,opencv)