6个案例手把手教你用Python和OpenCV进行图像处理
01 图像的读取、显示和保存
OpenCV提供了cv2模块,用于进行图像的处理操作。
1. 读取图像
OpenCV提供了cv2.imread()函数用于进行图像的读取操作。该函数的基本格式为:
retval = cv2.imread(filename[, flags])
其中:
- retval是返回值,其值是读取到的图像。
- filename是要读取图像的完整文件名。
- flags是读取标记,用来控制读取文件的类型。部分常用的标记值如表3-1所示,其中第一列的值与第三列的数值表示的含义一致。
▲表3-1 常用flags标记值
- 例3-1 使用cv2.imread()函数读取一幅图像
代码如下:
import cv2 as cv image = cv2.imread("F:/picture/lena.png") # 读取lena图像 print(image)
运行代码会得到lena图像的像素值,如图3-3所示。
▲图3-3 lena图像部分像素值
2. 显示图像
OpenCV提供了多个与图像显示有关的函数,下面简单介绍常用的几个。
namedWindow()函数用来创建指定的窗口,一般格式如下:
None = cv2.namedWindow(window)
其中,window是窗口的名字。例如:
cv2.namedWindow("image")
这句程序会新建一个名字为image的窗口。
imshow()函数用来显示图像,其一般格式如下:
None = cv2.imshow(window, image)
其中:
- window是窗口的名字。
- image是要显示的图像。
waitKey()函数用来等待按键,当有键被按下时,该语句会被执行。其一般格式如下:
retval= cv2.waitKey([delay])
其中:
- retval是返回值。
- delay表示等待键盘触发的时间,单位是ms。当该值为负数或0时表示无限等待,默认值为0。
destroyAllWindows函数用来释放所有窗口,其一般格式为:
None = cv2. destroyAllWindows ()
- 例3-2 显示读取的图像
代码如下:
import cv2 as cv # 导入从cv2模块 image = cv.imread("F:/picture/lena.png") # 读取lena图像 cv.namedWindow("image") # 创建一个image的窗口 cv.imshow("image", image) # 显示图像 cv.waitKey() # 默认为0,无限等待 cv.destroyAllWindows() # 释放所有窗口
程序运行结果如图3-4所示。
▲图3-4 例3-2的运行结果
3. 保存图像
OpenCV中提供了cv2.imwrite()函数来保存图像,其一般格式为:
retval= cv2.imwrite(filename, img[, params])
其中:
- retval是返回值。
- filename是要保存的图像的完整路径名,包括文件的扩展名。
- img是要保存的图像的名字。
- params是保存的类型参数,可选。
- 例3-3 编写程序,将读取到的图像保存
代码如下:
import cv2 as cv # 导入从cv2模块 image = cv.imread("F:/picture/lena.png") # 读取lena图像 cv.imwrite("F:/picture/lenaresult.png",image) #将图像保存到F:/picture/下,名字为lenaresult
02 图像通道的基本操作
在图像处理过程中,有时会根据需要对通道进行拆分与合并。在OpenCV中提供了split()和merge()函数对图像进行拆分与合并。下面对这两个函数进行介绍。
1. split()拆分函数
函数split()可以拆分图像的通道,例如BGR图像的三个通道。其一般格式如下:
b,g,r = cv2.split(img)
其中:
- b、g、r分别是B通道、G通道、R通道的图像信息。
- img是要拆分的图像。
- 例3-4 编写程序,使用split()函数对图像进行拆分
代码如下:
import cv2 as cv image = cv.imread("F:/picture/lena.png") b,g,r = cv.split(image) # 拆分图像通道分为b,g,r三个通道 cv.imshow("b",b) # 显示b通道的图像信息 cv.imshow("g",g) # 显示g通道的图像信息 cv.imshow("r",r) # 显示r通道的图像信息 cv.imshow("image", image) cv.waitKey() cv.destroyAllWindows()
程序运行结果如图3-5所示。
▲图3-5 例3-4的运行结果:a)原始图像,b)B通道图像,c)G通道图像,d)R通道图像
其中,图3-5a是原图,图3-5b是B通道的图像,图3-5c是G通道的图像,图3-5d是R通道的图像。
2. merge()合并函数
通道合并是通道拆分的逆过程,可以将三个通道的灰度图像合并为一张彩色图像。OpenCV中提供了merge()函数来实现图像通道的合并,其基本格式为:
imagebgr = cv2.merge([b,g,r])
其中:
- imagebgr是合并后的图像。
- b、g、r分别是B通道、G通道、R通道的图像信息。
- 例3-5 编写程序,演示合并图像的过程
代码如下:
import cv2 as cv image = cv.imread("F:/picture/lena.png") b,g,r = cv.split(image) # 拆分图像通道分为b,g,r三个通道 imagebgr = cv.merge([b,g,r]) # 将b,g,r三个通道的图像合并 cv.imshow("image", image) cv.imshow("imagegbgr", imagebgr) cv.waitKey() cv.destroyAllWindows()
程序运行结果如图3-6所示。
▲图3-6 例3-5的运行结果:a)原始图像 b)拆分并合并后的图像
其中,图3-6a是原始图像,图3-6b是经过拆分后又合并的图像。
03 图像属性的获取
在进行图像处理时经常需要获取图像的大小、类型等属性信息。下面介绍几个常用属性。
- shape:表示图像的大小。如果是彩色图像,则返回包含行数、列数和通道数的数组;如果是二值图像或灰度图像,则返回包含行数和列数的数组。
- size:表示返回的图像的像素数目。
- dtype:表示返回的图像的数据类型。
- 例3-6 编写程序,观察图像的属性值
代码如下:
import cv2 as cv image = cv.imread("F:/picture/lena.png") print("image.shape",image.shape) # 输出图像的大小属性 print("image.size",image.size) # 输出图像的像素数目属性 print("image.dtype",image.dtype) # 输出图像的类型属性
程序运行结果为:
image.shape (512, 512, 3) image.size 786432 image.dtype uint8
关于作者:高敬鹏,博士学历,硕士生导师,2002年至今,任职于哈尔滨工程大学信息与通信工程学院。研究方向主要包括人工智能、机器学习、图像处理、信号检测、目标识别、现代通信技术与电子系统等。
江志烨,博士学历,研究员,任职于北京航天长征飞行器研究所。