首页 > 脚本专栏 > python > python opencv与PIL图片读取保存及转化

python中opencv与PIL图片读取保存及相互转化方式

2023-09-08 09:00:13 作者：木南成长之路

这篇文章主要介绍了python中opencv与PIL图片读取保存及相互转化方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

opencv与PIL图片读取保存及相互转化

1. opencv读取显示图片

import cv2
img = cv2.imread('img.jpg')
cv2.imshow('img', img)  # 图片显示
cv2.waitKey()  # 括号内的输入控制imshow的持续时间

其中waitKey()表示等待用户按键，若不加这句，则图片显示一闪而过。

cv2读取的图片即img为numpy.ndarray，大小即为(w,h,c)。

2. PIL读取显示图片

from PIL import Image
image = Image.open("img.jpg")
image.show()

不需要waitKey()，图片会持续显示。

image为PIL格式。

3. opencv与PIL相互转换

PIL转opencv

import cv2
from PIL import Image
import numpy
image = Image.open("img.jpg")  # PIL读取
img = cv2.cvtColor(numpy.asarray(image), cv2.COLOR_RGB2BGR)  # 转换代码
cv2.imshow('img', img)  # opencv显示
cv2.waitKey()

先将PIL格式转化为numpy.ndarray，再转换成BGR通过opencv显示，opencv用的都是BGR格式。

opencv转PIL

import cv2
from PIL import Image
img = cv2.imread('img.jpg')  # opencv读取
image = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))  # 转换代码
image.show()  # PIL显示

先将numpy由BGR转成RGB，再转成PIL格式显示

4. 保存图片

opencv保存–imwrite()

import cv2
img = cv2.imread('img.jpg')
cv2.imwrite('XXX.jpg', img)  # 保存为XXX.jpg

PIL保存–save()

from PIL import Image
image = Image.open("img.jpg")
image.save('XXX.jpg')  # 保存为XXX.jpg

PIL图片与OpenCV图片的相互转换

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 方法进行转换。

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。