PIL
图片与OpenCV
图片的相互转换是通过数组格式转换的。通过type()
函数,我们可以观察到通过cv2.imread()
函数得到的图片格式为numpy.ndarray
;通过Image.open()
函数得到的图片格式为PIL.JpegImagePlugin.JpegImageFile
,而在Image
中存在通过数组格式加载图像的方法Image.fromarray
。因此,以数组格式作为中间者可实现PIL
图片与OpenCV
图片之间的相互转换,代码如下:
import cv2
import numpy as np
from PIL import Image
def load_img_by_cv(file):
return cv2.imread(file)
def load_img_by_PIL(file):
return Image.open(file)
def cv2PIL(img_cv):
return Image.fromarray(cv2.cvtColor(img_cv,cv2.COLOR_BGR2RGB))
def PIL2cv(img_pil):
return cv2.cvtColor(np.asarray(img_pil),cv2.COLOR_RGB2BGR)
注意
通过cv2
加载的图片,默认色道格式为BGR
,可通过cv2.cvtColor
函数进行转换;
通过PIL
加载的图片,默认色道格式为RGB
,可通过图像的convert
方法进行转换。