算法统一图片输入格式RGB/BGR

统一图片输入格式

问题

在做一些图像处理项目时候，有些模型对输入图片格式有一定的要求，不一致的图片格式输入进去可能会造成一些错误情况出现；统一输入图片读取后的格式是非常有必要的；

在opencv中，使用cv2.imread读取jpg类型图片，得到的图像矩阵形式是BGR格式； 对于png类型图片，得到的图像矩阵形式是RBG格式；如果算法处理的图片既有jpg又有png，那么很有必要在算法入口处对两种图片得到的图像矩阵格式进行统一

方法

假设这里算法需要的是rgb类型格式的图片矩阵， 传入的是本地图片的名称img_name。

• 方法1， 根据图像名称以及cv2.cvtColor统一格式

import cv2


img = cv2.imread(image_name)
if image_name.endswith('jpg'):  # 如果是jpg，需要进行转换
    img = cv2.cvtColo(img, cv2.COLOR_BGR2RGB)

• 方法2， 使用图像处理库PIL来读取图片

对于彩色图像，不管其图像格式是PNG，还是BMP，或者JPG，在PIL中，使用Image模块的open()函数打开后，返回的图像对象的模式都是RGB。

from PIL import Image

img = Image.open(image_name)

参考

• https://blog.youkuaiyun.com/icamera0/article/details/50843172
• https://stackoverflow.com/questions/59581538/how-to-check-whether-my-image-is-rgb-format-or-bgr-format-in-python-how-do-i-co