计算机视觉
计算机视觉(Computer Vision)是人工智能的一个重要分支,它研究如何让计算机“看”和“理解”图像和视频。也叫机器视觉(Machine Vision)。
早在深度学习之前,人们就开始探索图像的处理了,OpenCV 诞生于深度学习之前,是功能最全面的开源库。它提供了从最基础的图像读写、处理到复杂的特征匹配、目标跟踪、校准等一系列工具。其核心优势在于传统的计算机视觉算法,如图像处理、滤波、形态学操作等。
它向前承接传统以数组为单位的图像处理,向后允许你加载一些现成的算法、模型,是传统图像处理与深度学习之间重要的桥梁。
安装
opencv(opensource computer vision)开源计算机视觉库,包含大量的图像处理函数。
opencv-python需要根据环境选择正确的包
有4个不同的包,你只能选择其中的一个。不要在同一环境中安装多个不同的包。
所有包都使用相同的名称空间(cv2
)。如果你在同一环境中安装了多个不同的包,请使用 pip uninstall
卸载所有包,然后只重新安装一个包。
a. 针对 标准桌面环境的包(适用于 Windows、macOS、几乎所有 GNU/Linux 发行版)
- 选项 1 - 主模块包:
pip install opencv-python
- 选项 2 - 完整包(包含主模块和 contrib/extra 模块):
pip install opencv-contrib-python
(请参考 OpenCV 文档 中的 contrib/extra 模块列表)
b. 针对服务器(无窗口)环境的包(例如 Docker、云环境等),无 GUI 库依赖
这些包比前面两种包更小,因为它们不包含任何图形用户界面功能(未编译 Qt 或其他 GUI 组件)。这意味着这些包避免了对 X11 库的庞大依赖,从而例如可以生成更小的 Docker 镜像。如果你不使用 cv2.imshow
等函数,或者你使用其他包(例如 PyQt)来构建界面而非 OpenCV,则应始终使用这些包。
- 选项 3 - 无窗口主模块包:
pip install opencv-python-headless
- 选项 4 - 无窗口完整包(包含主模块和 contrib/extra 模块):
pip install opencv-contrib-python-headless
(请参考 OpenCV 文档 中的 contrib/extra 模块列表)
OpenCv作为开源软件,自然有大量的教程,我读过一些纸质书籍,也看过一些开源教程,总体来说,对于入门与进阶来说,需要的是细致的基础讲解、完整的处理流程,非常推荐官方的OpenCV-Python教程。
图像读取与显示
图像读取
函数签名:cv2.imread(filename, flags=cv2.IMREAD_COLOR) -> image
参数说明:
filename
:要读取的文件名flags
:读取标志,默认值为cv2.IMREAD_COLOR,表示读取为BGR格式。可选参数:- cv2.IMREAD_COLOR:读取为BGR格式,默认值。对应数值为1。
- cv2.IMREAD_GRAYSCALE:读取为灰度图,对应数值为0。
- cv2.IMREAD_UNCHANGED:读取为原图,包含alpha通道,对应数值为-1。
返回值:
- 返回读取的图像对象, 如果读取失败, 不报错,返回None
-
CV2默认使用系统编码,如果文件路径含中文无法正常读取,需要把系统编码改为
utf-8
。
图像对象的常用属性:
shape
:图像的形状,(height, width, channels) 其中channels为3、4,表示通道数为RGB、RGBA(A是透明度),如果是灰度、二值图,通道数为1的情况会只返回(height, width)。size
:图像的像素数,height * width * channelsdtype
:图像的数据类型ndim
:图像的维度itemsize
:图像的元素大小nbytes
:图像的内存大小
import cv2
path= "10.jpg"
img = cv2.imread(path)
print(img.shape)
图像保存
函数签名:cv2.imwrite(filename, img, [params])
参数说明:
filename
:要保存的文件名img
:要保存的图像对象(数据)params
:可选参数,用于指定保存的格式和质量,可选参数:- cv2.IMWRITE_JPEG_QUALITY:JPEG图像质量,范围为0-100,默认值为95
- cv2.IMWRITE_PNG_COMPRESSION:PNG图像压缩级别,范围为0-9,默认值为3
- cv2.IMWRITE_PXM_BINARY:PPM/PGM/PBM图像格式,默认值为False
import cv2
path= "10.jpg"
img = cv2.imread(path)
cv2.imwrite('new_img.jpg',img, [cv2.IMWRITE_JPEG_QUALITY, 100])
图像显示、等待按键、关闭窗口
cv2.imshow函数签名:cv2.imshow(winname, img)
参数说明:
winname
:窗口名称img
:要显示的图像对象
cv2.waitKey函数签名:cv2.waitKey([delay]) -> retval
参数说明:
delay
:等待时间,单位为毫秒。当值为0时,表示无限等待。
返回值:
- 返回按键的ASCII码(大小写敏感),如果到时间没有按键按下,则返回-1。
注意:
- 如果设置了中文输入法,键盘的捕获会被输入法占用,导致无法正常等待按键按下。
- 如果弹出多个相同名称的窗口,只会显示一个窗口,窗口内的图像内容会被覆盖。
上面两个函数是成对出现的,cv2.imshow
用于显示图像,cv2.waitKey
用于等待按键按下或者到时间自动关闭窗口。
下面两个函数都表示关闭窗口,区别在于destroyWindow
是关闭指定窗口,而destroyAllWindows
是关闭所有窗口。
destroyWindow函数签名:cv2.destroyWindow(winname)
参数说明:
winname
:要关闭的窗口名称
destroyAllWindows函数签名:cv2.destroyAllWindows()
import cv2
path= "10.jpg"
img = cv2.imread(path)
# 显示图片, 参数:img: 图像对象, 'Image': 窗口名称
# 如果窗口名称重复,则不会创建新的窗口,而是将图像显示在已有的窗口中
cv2.imshow('Image', img)
# 等待按键按下,0表示无 限等待,其他数字表示等待时间(毫秒)
# 这个方法会阻塞程序,直到按键按下,返回值为按键的ASCII码。如果设置了中文输入法,键盘的捕获会被输入法占用,导致无法正常等待按键按下。
key = cv2.waitKey(0)
# 如果按键为q,则退出
if key == ord('q'):
# 关闭指定窗口
cv2.destroyWindow('Image')
# 关闭所有窗口
cv2.destroyAllWindows()
动态图像
视频读取与显示
# cv2.VideoCapture(参数):参数可以是整数、字符串、流媒体
# 整数:表示摄像头的编号
# 0:表示默认摄像头
# 1:表示第二个摄像头
# 字符串:表示视频文件的路径(绝对相对都可以):1.mp4
# 流媒体:表示视频流: http://192.168.1.100:8080
cap = cv2.VideoCapture(0) # 摄像头的语柄
if not cap.isOpened():
print("Error: Could not open video.")
cap.release() # 释放摄像头
while True:
ret,frame = cap.read() # 读取视频帧
# ret:表示是否读取成功(True/False)
# frame:表示读取到的视频帧数据(numpy数组)
if not ret:
print("Error: Could not read frame.")
break
cv2.imshow("frame",frame) # 显示视频帧。同个窗口名称,会覆盖窗口内容,而不是新开窗口。
if cv2.waitKey(1) == ord('q'): # 按q键退出循环
break
cap.release() # 释放摄像头
cv2.destroyAllWindows() # 关闭所有窗口
视频保存
函数签名:cv2.VideoWriter(filename, fourcc, fps, frameSize)
参数说明:
- filename:视频文件名
- fourcc:视频编码格式
- fps:帧率
- frameSize:帧大小
import cv2
import numpy as np
# cv2.VideoCapture(参数):参数可以是整数、字符串、流媒体
# 整数:表示摄像头的编号
# 0:表示默认摄像头
# 1:表示第二个摄像头
# 字符串:表示视频文件的路径(绝对相对都可以):1.mp4
# 流媒体:表示视频流:rtsp://192.168.1.100:8554/test 或者 http://192.168.1.100:8080
cap = cv2.VideoCapture(0) # 摄像头的语柄
# 视频的编码格式
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
# 视频的输出路径
out = cv2.VideoWriter('output.mp4', fourcc, 20.0, (640,480))
if not cap.isOpened():
print("Error: Could not open video.")
cap.release() # 释放摄像头
while True:
ret,frame = cap.read() # 读取视频帧
# ret:表示是否读取成功(True/False)
# frame:表示读取到的视频帧数据(numpy数组)
if not ret:
print("Error: Could not read frame.")
break
out.write(frame) # 写入视频帧
cv2.imshow("frame",frame) # 显示视频帧。同个窗口名称,会覆盖窗口内容,而不是新开窗口。
if cv2.waitKey(1) == ord('q'): # 按q键退出循环
break
out.release() # 释放视频
cap.release() # 释放摄像头
cv2.destroyAllWindows() # 关闭所有窗口
图像绘制
绘制直线、矩形
line
函数签名:cv2.line(img, pt1, pt2, color, thickness, lineType, shift)
rectangle
函数签名:cv2.rectangle(img, pt1, pt2, color, thickness, lineType, shift)
参数说明:
img
:图像对象pt1
:起点坐标pt2
:终点坐标color
:颜色thickness
:线宽lineType
:线型shift
:位移,默认为0,表示坐标是整数,如果大于1,则表示坐标是浮点数,例如shift=2,则表示坐标是浮点数。可用于实现亚像素精度绘图。-
对于矩形来说,
pt1
是左上角,pt2
是右 下角。
圆形、椭圆
circle
函数签名:cv2.circle(img, center, radius, color, thickness, lineType, shift)
ellipse
函数签名:cv2.ellipse(img, center, axes, angle, startAngle, endAngle, color, thickness, lineType, shift)
参数说明:
img
:图像对象center
:圆心坐标radius
:半径axes
:椭圆长轴和短轴长度angle
:椭圆旋转角度startAngle
:椭圆起始角度endAngle
:椭圆结束角度color
:颜色thickness
:线宽lineType
:线型shift
:位移,默认为0,表示坐标是整数,如果大于1,则表示坐标是浮点数,例如shift=2,则表示坐标是浮点数。可用于实现亚像素精度绘图。
多边形
polylines函数签名:cv2.polylines(img, pts, isClosed, color, thickness, lineType, shift)
参数说明:
img
:图像对象pts
:顶点列表isClosed
:是否闭合color
:颜色thickness
:线宽lineType
:线型shift
:位移,默认为0,表示坐标是整数,如果大于1,则表示坐标是浮点数,例如shift=2,则表示坐标是浮点数。可用于实现亚像素精度绘图。
# 导入库
import cv2
# 读取图片
img = cv2.imread('10.jpg')
# 绘制直线
# 参数:img: 图像对象, (100, 100): 起点坐标, (200, 200): 终点坐标, (0, 0, 255): 颜色, 2: 线宽
cv2.line(img, (100, 100), (200, 200), (0, 0, 255), 2)
# 绘制矩形
# 参数:img: 图像对象, (100, 100): 左上角坐标, (200, 200): 右下角坐标, (0, 0, 255): 颜色, 2: 线宽
cv2.rectangle(img, (100, 100), (200, 200), (0, 0, 255), 2)
# 绘制圆形
# 参数:img: 图像对象, (150, 150): 圆心坐标, 50: 半径, (0, 0, 255): 颜色, 2: 线宽
cv2.circle(img, (150, 150), 50, (0, 0, 255), 2)
# 绘制多边形
# 参数:img: 图像对象, [pts]: 顶点列表, True: 是否闭合, (0, 0, 255): 颜色, 2: 线宽
pts = np.array([[100, 100], [200, 100], [200, 200], [100, 200]])
cv2.polylines(img, [pts], True, (0, 0, 255), 2)
# 绘制椭圆
# 参数:img: 图像对象, (150, 150): 椭圆中心坐标, (100, 50): 椭圆长轴和短轴长度, 0: 椭圆旋转角度, 0: 椭圆起始角度, 360: 椭圆结束角度, (0, 0, 255): 颜色, 2: 线宽
cv2.ellipse(img, (150, 150), (100, 50), 0, 0, 360, (0, 0, 255), 2)
# 显示图片
cv2.imshow('Image', img)
# 等待按键按下
cv2.waitKey(0)
# 关闭所有窗口
cv2.destroyAllWindows()
绘制文字
函数签名:cv2.putText(img, text, org, fontFace, fontScale, color, thickness, lineType, bottomLeftOrigin)
参数说明:
img
:图像对象text
:要绘制的文字org
:文字的左下角坐标fontFace
:字体类型fontScale
:字体缩放比例(大于1时,文字会变大、0~1时,文字会变小,小于0时,文本将倒置)color
:文字颜色thickness
:文字粗细lineType
:线型bottomLeftOrigin
:是否从左下角开始绘制
函数直接作用于图像对象,不需要返回值。
import cv2
# 读取图片
img = cv2.imread('10.jpg')
# 绘制文字:Hello, World! 位置为(100, 100),字体为cv2.FONT_HERSHEY_SIMPLEX,字体缩放比例为1,颜色为红色,粗细为2
cv2.putText(img, 'Hello, World!', (100, 100), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2)
# 显示图片
cv2.imshow('Image', img)
# 展示1秒
cv2.waitKey(1000)
# 关闭所有窗口
cv2.destroyAllWindows()
PIL库绘制中文
cv2的在图片上的绘制语法主要缺陷为无法绘制中文,需要使用PIL库来绘制中文。
字体文件:一种可以把码点转换为像素显示的超大型字典。
import cv2
import numpy as np
from PIL import Image, ImageDraw, ImageFont
# 打开图片
img = Image.open('imgs/img/10.jpg') # 替换为你的图片路径
# 设置字体(确保路径正确,Windows下常见字体路径如下)
font_path = "simhei.ttf" # 黑体
font_size = 60
font = ImageFont.truetype(font_path, font_size)
# 要绘制的文字
text = "机器视觉"
# 创建绘图对象
draw = ImageDraw.Draw(img)
# 计算文字尺寸
text_width, text_height = draw.textsize(text, font=font)
# 计算图片中心
img_width, img_height = img.size
x = (img_width - text_width) // 2
y = (img_height - text_height) // 2
# 绘制文字(可设置颜色和描边等)
draw.text((x, y), text, font=font, fill=(255, 0, 0)) # 红色
# 保存或显示图片
img.show()
# img.save('output.jpg')
图像运算
加减法的前提是两张图片的尺寸相同。
图像加法、图像减法
import cv2
import numpy as np
path = "xxx.bmp"
def draw_add(path):
img = cv2.imread(path)
img_height, img_width = img.shape[:2]
# 创建一个全白的mask_img
mask_img = np.zeros((img_height, img_width, 3), dtype=np.uint8) + 255
# 在mask_img上写上Showcase
cv2.putText(mask_img, "Showcase", (img_width//4, img_height//2), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 0), 5)
# 将mask_img和img相加
img3 = cv2.add(img, mask_img)
cv2.imwrite(f"add_{path}", img3)
def draw_subtract(path):
img = cv2.imread(path)
img_height, img_width = img.shape[:2]
# 创建一个全白的mask_img
mask_img = np.zeros((img_height, img_width, 3), dtype=np.uint8) + 255
# 在mask_img上画一个圆
cv2.circle(mask_img, (img_width//2, img_height//2), int(min(img_width, img_height)*0.45), (0, 0, 0), -1)
# 将mask_img和img相减
img3 = cv2.subtract(img, mask_img)
cv2.imwrite(f"subtract_{path}", img3)
draw_add(path)
draw_subtract(path)
图像加权cv2.addWeighted
函数签名:cv2.addWeighted(src1, alpha, src2, beta, gamma) -> dst
参数说明:
src1
:图像1alpha
:图像1的强度,范围为0-1src2
:图像2beta
:图像2的强度,范围为0-1gamma
:偏移量(即图像在加权后,整体增加的值,一般为0)
图像1和图像2的强度不需要合起来等于1,可以大于1,也可以小于1。
import cv2
import numpy as np
img1 = cv2.imread("green.png")
img2 = cv2.imread("red.png")
img3 = cv2.addWeighted(img1, 1, img2, 1,0)
cv2.imshow("img4", img3)
cv2.waitKey(0)
基础图像操作
图像缩放
函数签名:cv2.resize(src, dsize, fx, fy, interpolation) -> dst
参数说明:
src
:源图像dsize
:目标图像大小,如果为None,则根据fx和fy计算,传入的是(width, height)fx
:水平缩放比例fy
:垂直缩放比例interpolation
:插值方式,默认值为cv2.INTER_LINEAR:双线性插值
当resize指定的新尺寸大于原尺寸时,会使用插值方式来填充新产生的像素。
import cv2
import numpy as np
img1 = cv2.imread("xxx.png")
# 固定值缩放
img2 = cv2.resize(img1, (100, 100))
# 按比例缩放
img2 = cv2.resize(img1, dsize = None, fx = 0.5, fy = 0.5)
拆分与合并通道
拆分通道函数签名:cv2.split(img) -> list
合并通道函数签名:cv2.merge(list) -> img
参数说明:
img
:图像对象
返回值:
list
:不同通道的图像对象
import cv2
import numpy as np
img = cv2.imread("xxx.png")
b, g, r = cv2.split(img)
img2 = cv2.merge([b, g, r])
cv2.imshow("img", img)
cv2.imshow("img2", img2)
cv2.waitKey(0)
cv2.destroyAllWindows()
图像翻转
函数签名:cv2.flip(src, flipCode) -> dst
参数说明:
src
:源图像flipCode
:翻转方式,可选参数:- 0:上下翻转(垂直翻转)
- 1:左右翻转(水平翻转)
- -1:上下左右翻转(垂直水平翻转)
返回值:
- 翻转后的图像对象
import cv2
img = cv2.imread("xxx.png")
# 上下翻转(垂直翻转)
img_vertical = cv2.flip(img, 0)
# 左右翻转(水平翻转)
img_horizontal = cv2.flip(img, 1)
# 上下左右翻转(垂直水平翻转)
img_both = cv2.flip(img, -1)
cv2.imshow("Original", img)
cv2.imshow("Vertical Flip", img_vertical)
cv2.imshow("Horizontal Flip", img_horizontal)
cv2.imshow("Both Flip", img_both)
cv2.waitKey(0)
cv2.destroyAllWindows()
图像裁剪
图像裁剪使用NumPy数组切片操作,不需要专门的OpenCV函数。
操作语法:img[start_y:end_y, start_x:end_x]
参数说明:
start_y
:裁剪区域的起始行(y坐标)end_y
:裁剪区域的结束行(y坐标)start_x
:裁剪区域的起始列(x坐标)end_x
:裁剪区域的结束列(x坐标)
注意:坐标系以图像左上角为原点,y轴向下,x轴向右
import cv2
img = cv2.imread("xxx.png")
# 获取图像尺寸
height, width = img.shape[:2]
# 裁剪中心区域(从坐标(100,100)到(300,300))
cropped_img = img[100:300, 100:300]
# 裁剪图像上半部分
top_half = img[0:height//2, 0:width]
# 裁剪图像右下角四分之一
bottom_right = img[height//2:height, width//2:width]
cv2.imshow("Original", img)
cv2.imshow("Cropped", cropped_img)
cv2.imshow("Top Half", top_half)
cv2.imshow("Bottom Right", bottom_right)
cv2.waitKey(0)
cv2.destroyAllWindows()
图像平移
函数签名:cv2.warpAffine(src, M, dsize, flags, borderMode, borderValue) -> dst
参数说明:
src
:源图像M
:2×3的仿射变换矩阵dsize
:输出图像的大小,格式为(width, height)flags
:插值方法,默认为cv2.INTER_LINEARborderMode
:边界处理方式,默认为cv2.BORDER_CONSTANTborderValue
:边界填充值,默认为0
变换矩阵格式:[[1, 0, dx], [0, 1, dy]]
,其中dx为x方向位移,dy为y方向位移
import cv2
import numpy as np
img = cv2.imread("xxx.png")
height, width = img.shape[:2]
# 向右平移100像素,向下平移50像素
M = np.float32([[1, 0, 100], [0, 1, 50]])
translated_img = cv2.warpAffine(img, M, (width, height))
# 向左平移50像素,向上平移30像素
M2 = np.float32([[1, 0, -50], [0, 1, -30]])
translated_img2 = cv2.warpAffine(img, M2, (width, height))
cv2.imshow("Original", img)
cv2.imshow("Translated Right-Down", translated_img)
cv2.imshow("Translated Left-Up", translated_img2)
cv2.waitKey(0)
cv2.destroyAllWindows()
图像旋转
旋转矩阵函数签名:cv2.getRotationMatrix2D(center, angle, scale) -> M
参数说明:
center
:旋转中心点坐标,格式为(x, y)angle
:旋转角度(度数),正值表示逆时针旋转scale
:缩放比例,1.0表示不缩放
应用旋转需要配合cv2.warpAffine
函数使用。
import cv2
import numpy as np
img = cv2.imread("xxx.png")
height, width = img.shape[:2]
# 以图像中心为旋转点,逆时针旋转45度,不缩放
center = (width // 2, height // 2)
M = cv2.getRotationMatrix2D(center, 45, 1.0)
rotated_img = cv2.warpAffine(img, M, (width, height))
# 以图像中心为旋转点,顺时针旋转30度,放大1.5倍
M2 = cv2.getRotationMatrix2D(center, -30, 1.5)
rotated_scaled_img = cv2.warpAffine(img, M2, (width, height))
cv2.imshow("Original", img)
cv2.imshow("Rotated 45°", rotated_img)
cv2.imshow("Rotated -30° Scaled 1.5x", rotated_scaled_img)
cv2.waitKey(0)
cv2.destroyAllWindows()
仿射变换
仿射变换矩阵函数签名:cv2.getAffineTransform(src, dst) -> M
参数说明:
src
:源图像中的三个点坐标,numpy数组格式dst
:目标图像中对应的三个点坐标,numpy数组格式
返回值 :
- 2×3的仿射变换矩阵
仿射变换能够实现旋转、平移、缩放和错切等变换的组合。
import cv2
import numpy as np
img = cv2.imread("xxx.png")
height, width = img.shape[:2]
# 定义源图像中的三个点(左上角、右上角、左下角)
pts1 = np.float32([[0, 0], [width-1, 0], [0, height-1]])
# 定义目标图像中对应的三 个点(实现错切变换)
pts2 = np.float32([[0, height*0.33], [width*0.85, height*0.25], [width*0.15, height*0.7]])
# 计算仿射变换矩阵
M = cv2.getAffineTransform(pts1, pts2)
# 应用仿射变换
affine_img = cv2.warpAffine(img, M, (width, height))
# 在原图上标记三个关键点
cv2.circle(img, (0, 0), 5, (255, 0, 0), -1)
cv2.circle(img, (width-1, 0), 5, (255, 0, 0), -1)
cv2.circle(img, (0, height-1), 5, (255, 0, 0), -1)
cv2.imshow("Original", img)
cv2.imshow("Affine Transform", affine_img)
cv2.waitKey(0)
cv2.destroyAllWindows()
透视变换
透视变换矩阵函数签名:cv2.getPerspectiveTransform(src, dst) -> M
透视变换函数签名:cv2.warpPerspective(src, M, dsize, flags, borderMode, borderValue) -> dst
参数说明:
src
:源图像中的四个点坐标,numpy数组格式dst
:目标图像中对应的四个点坐标,numpy数组格式M
:3×3的透视变换矩阵dsize
:输出图像的大小,格式为(width, height)
透视变换能够模拟三维空间中的投影效果,常用于文档矫正、鸟瞰图变换等。
import cv2
import numpy as np
img = cv2.imread("xxx.png")
height, width = img.shape[:2]
# 定义源图像中的四个点(矩形的四个角)
pts1 = np.float32([[0, 0], [width, 0], [0, height], [width, height]])
# 定义目标图像中对应的四个点(梯形效果)
pts2 = np.float32([[0, 0], [width, 0], [width*0.3, height], [width*0.7, height]])
# 计算透视变换矩阵
M = cv2.getPerspectiveTransform(pts1, pts2)
# 应用透视变换
perspective_img = cv2.warpPerspective(img, M, (width, height))
cv2.imshow("Original", img)
cv2.imshow("Perspective Transform", perspective_img)
cv2.waitKey(0)
cv2.destroyAllWindows()
图像扩边
函数签名:cv2.copyMakeBorder(src, top, bottom, left, right, borderType, value) -> dst
参数说明:
src
:源图像top
:上边框宽度(像素)bottom
:下边框宽度(像素)left
:左边框宽度(像素)right
:右边框宽度(像素)borderType
:边框类型,常用类型:- cv2.BORDER_CONSTANT:常数边框,需要指定value参数
- cv2.BORDER_REFLECT:反射边框
- cv2.BORDER_REPLICATE:复制边框
- cv2.BORDER_WRAP:包装边框
value
:边框颜色值(当borderType为BORDER_CONSTANT时使用)
import cv2
img = cv2.imread("xxx.png")
# 添加常数边框(黑色)
border_constant = cv2.copyMakeBorder(img, 50, 50, 50, 50, cv2.BORDER_CONSTANT, value=(0, 0, 0))
# 添加常数边框(红色)
border_red = cv2.copyMakeBorder(img, 30, 30, 30, 30, cv2.BORDER_CONSTANT, value=(0, 0, 255))
# 添加反射边框
border_reflect = cv2.copyMakeBorder(img, 50, 50, 50, 50, cv2.BORDER_REFLECT)
# 添加复制边框
border_replicate = cv2.copyMakeBorder(img, 50, 50, 50, 50, cv2.BORDER_REPLICATE)
# 添加包装边框
border_wrap = cv2.copyMakeBorder(img, 50, 50, 50, 50, cv2.BORDER_WRAP)
cv2.imshow("Original", img)
cv2.imshow("Constant Border (Black)", border_constant)
cv2.imshow("Constant Border (Red)", border_red)
cv2.imshow("Reflect Border", border_reflect)
cv2.imshow("Replicate Border", border_replicate)
cv2.imshow("Wrap Border", border_wrap)
cv2.waitKey(0)
cv2.destroyAllWindows()
色彩处理
科普:什么是HSV?
HSV:相比RGB相比,HSV能更好的表示同个颜色的不同值(饱和度和明度)
- H: 色调,取值范围是0-180,表示颜色
- S: 饱和度,取值范围是0-255,表示颜色的纯度
- V: 明度,取值范围是0-255,表示颜色的亮度
什么是24位真彩屏?
每个像素点的取值返回是0~255,即在二进制中需要8位来表示(2的8次方)
那么 R G B 三个通道就需要 8 + 8 + 8 = 24 位来表示
难道还有假彩屏?
是的,因为人眼并不能分辨出 255 * 255 * 255 种颜色,所以可以适当的减少一些颜色,也不影响人眼感知
所以就有了16bit屏幕(单片机常用)?
灰度图
灰度图 适用于图像处理,如边缘检测及其应用:图像分割、轮廓检测 只有1个通道,取值范围是0-255,表示颜色的亮度
import numpy as np
import cv2
# 方式1 (支持自定义权重)
img = cv2.imread("xxx.png")
B,G,R = cv2.split(img)
gray = R*0.299 + G*0.587 + B*0.114
gray = gray.astype(np.uint8)
# 方式2 (最简单,最常用)
gray = cv2.imread("xxx.png",0)
# 方式3 (可以转化为各种色彩空间)
img = cv2.imread("xxx.png")
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
cv2.imshow("gray",gray)
cv2.waitKey(0)
cv2.destroyAllWindows()
二值化图
二值化图 适用于图像压缩 只有1个通道,取值范围是0或1
方法 | 核心思想 | 如何工作? | 优点 | 缺点 | 适用场景 |
---|---|---|---|---|---|
固定阈值法 | 设置一个固定的“分界线”来区分黑白。 | 你先设定一个固定的数值,比如 127。然后程序会遍历图像中的每一个像素点:如果像素值大于 127,就把它变成纯白色(255)。如果像素值小于或等于 127,就变成纯黑色(0)。最后,再把白色变为 1,黑色变为 0。 | 简单粗暴,速度快。 就像给所有学生设一条及格线,谁分数高谁就及格。 | 太死板,不灵活。 面对一张整体很暗的照片,大部分像素可能都低于 127,结果整个图都变成黑的了。对光线不均匀的图片完全无效。 | 适合处理光线均匀、对比度高的简单图像,比如扫描的黑白文档。 |
全局平均值法 | 自动计算图像的“平均分数”作为分界线。 | 程序会先计算整张图片的所有像素值的平均值,并把这个平均值作为动态的阈值。然后按照固定阈值法的方式进行二值化。 | 更智能。 它会根据图片本身的亮度来决定阈值,所以对整体偏亮或偏暗的图片效果更好。 | 忽略局部差异。 如果一张图片既有明亮的区域(比如太阳光),又有很暗的区域(比如阴影),这个方法就不太行了,因为它只用一个平均值来判断所有地方。 | 适合处理整体亮度统一,但亮度值不确定的图片。 |
自适应均值法 | “因地制宜”,为每个小区域设置不同的分界线。 | 程序会把图片分成很多个小格子。然后,它会只看每个格子里的像素,计算这个小区域的平均值,并把这个值作为这个小格子的阈值。 | 最灵活,效果好。 完美解决了局部光线不均的问题,比如在有阴影的纸张上,文字依然能被清晰地二值化。 | 计算量大。 因为要处理很多个小区域,所以比前两种方法慢一些。同时,你需要设置小格子的大小,这个参数的选择会影响最终效果。 | 广泛应用于文档识别、车牌识别、自动驾驶等对局部细节要求高的场景。 |
自适应高斯法 | “加权平均”,为每个小区域设置更智能的分界线。 | 类似于自适应均值法,也是将图像分成小格子。但不同的是,它会给离中心点更近的像素更高的权重,来计算区域的加权平均值,作为阈值。这样中心像素的影响更大。 | 相比自适应均值法,它对噪声的敏感度更低,能得到更平滑的二值化结果。 | 计算量比自适应均值法更大,因为要进行加权计算。 | 图像中有较多噪声或细节,需要更平滑二值化效果的场景。 |
Otsu's 方法 | 寻找一个最佳阈值,让图像的“黑白对比度”最大。 | 它会通过分析图像的灰度直方图(用来展示像素分布的图)来寻找最佳阈值。这个方法假设图像只有两类像素(前景和背景),然后找到一个阈值,让这两类像素之间的差异最大。 | 全自动,效果好。 无需手动设置任何参数,就能找到一个最优的阈值,特别适合那些背景和前景对比度很明显的图片。 | 有前 提限制。 如果图片的前景和背景混合得很复杂,或者灰度直方图不是明显的双峰形状,这个方法的表现就会变差。 | 适合处理对比度高的图片,特别是那些有清晰前景和背景的图片。 |
什么是直方图?
横坐标:表示的是像素的数值,最左边是0,最右边是255
纵坐标:表示的是像素的个数
常用于观察图片的像素分布
归一化(Normalization)是一种常见的数据预处理技术
用于改变数据的范围以使其落在一个特定的区间,通常是[0,1]或者[-1,1]。
归一化的目的是消除数据量纲和尺度的影响,使得不同尺度或单位的数据可以在同一水平上进行比较或处理。
import cv2
import numpy as np
import matplotlib.pyplot as plt
# 确保能找到图片,请将 'xxx.bmp' 替换为你自己的图片路径
IMAGE_PATH = 'xxx.bmp'
def show_all_images(original_img, binary_images, titles):
"""
使用 matplotlib 同时显示原始图像和所有二值化图像。
:param original_img: 原始灰度图像
:param binary_images: 二值化图像列表
:param titles: 图像标题列表
"""
num_images = len(binary_images) + 1 # +1 for original image
# 计算子图的行数和列数 (尽量接近正方形布局)
cols = int(np.ceil(np.sqrt(num_images)))
rows = int(np.ceil(num_images / cols))
fig, axes = plt.subplots(rows, cols, figsize=(15, 10))
fig.suptitle('Image Thresholding Comparison', fontsize=16)
# 如果只有一个子图,将axes转换为列表
if num_images == 1:
axes = [axes]
else:
axes = axes.flatten()
# 显示原始图像
axes[0].imshow(original_img, cmap='gray')
axes[0].set_title('Original Grayscale Image')
axes[0].axis('off')
# 显示二值化图像
for i, (binary_img, title) in enumerate(zip(binary_images, titles)):
axes[i + 1].imshow(binary_img, cmap='gray')
axes[i + 1].set_title(title)
axes[i + 1].axis('off')
# 隐藏多余的子图
for i in range(num_images, len(axes)):
axes[i].axis('off')
plt.tight_layout()
plt.show()
def get_binary_image(img, method, *args, **kwargs):
"""
根据不同方法获取二值化图像。
:param img: 原始灰度图像
:param method: 阈值处理方法,可选 'fixed', 'global_mean', 'adaptive_mean', 'adaptive_gaussian', 'otsu'
"""
if method == 'fixed':
# 固定阈值法
_, binary = cv2.threshold(img, 127, 255, cv2.THRESH_BINARY)
elif method == 'global_mean':
# 全局平均值法
_, binary = cv2.threshold(img, img.mean(), 255, cv2.THRESH_BINARY)
elif method == 'adaptive_mean':
# 自适应均值阈值法
binary = cv2.adaptiveThreshold(img, 255, cv2.ADAPTIVE_THRESH_MEAN_C,
cv2.THRESH_BINARY, 11, 2)
elif method == 'adaptive_gaussian':
# 自适应高斯阈值法
binary = cv2.adaptiveThreshold(img, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C,
cv2.THRESH_BINARY, 11, 2)
elif method == 'otsu':
# Otsu's 方法
_, binary = cv2.threshold(img, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)
else:
raise ValueError("Invalid method. Choose from 'fixed', 'global_mean', 'adaptive_mean', 'adaptive_gaussian', 'otsu'")
# 将二值图的像素值从 0/255 转换为 0/1,以便于分析和存储
return (binary / 255).astype(int)
def main():
"""主函数,演示不同二值化方法。"""
# 读取灰度图像
img = cv2.imread(IMAGE_PATH, cv2.IMREAD_GRAYSCALE)
if img is None:
print(f"Error: Could not read image at {IMAGE_PATH}")
return
# 准备各种二值化方法
methods = [
('fixed', 'Fixed Thresholding (127)'),
('global_mean', 'Global Mean Thresholding'),
('adaptive_mean', 'Adaptive Mean Thresholding'),
('adaptive_gaussian', 'Adaptive Gaussian Thresholding'),
('otsu', "Otsu's Thresholding")
]
# 生成所有二值化图像
binary_images = []
titles = []
for method, title in methods:
binary_image = get_binary_image(img, method)
binary_images.append(binary_image)
titles.append(title)
# 同时显示所有图像
show_all_images(img, binary_images, titles)
if __name__ == "__main__":
main()
形态学处理
膨胀和腐蚀
膨胀函数签名:cv2.dilate(img,kernel,iterations=5)
腐蚀函数签名:cv2.erode(img,kernel,iterations=5)
参数:
- img:输入图像
- kernel:膨胀和腐蚀的核
- iterations:膨胀和腐蚀的迭代次数
膨胀工作原理:类似卷积操作,将核覆盖在图像上,如果核的中心点在图像的范围内,则将核的中心点替换为核的最大值。
腐蚀工作原理与膨胀相反,将核覆盖在图像上,如果核的中心点在图像的范围内,则将核的中心点替换为核的最小值。
import cv2
import numpy as np
img = cv2.imread("dt2.png")
# 膨胀和腐蚀的操作
kernel = np.ones((3,3),np.uint8)
# 腐蚀
erode = cv2.erode(img,kernel,iterations=5)
# 膨胀
dilate = cv2.dilate(img,kernel,iterations=5)
cv2.imshow("img",img)
cv2.imshow("dilate",dilate)
cv2.waitKey(0)
cv2.destroyAllWindows()
- 开运算: 先腐蚀后膨胀,用于去除图像中的小噪声
- 闭运算: 先膨胀后腐蚀,用于填充图像中的小孔洞
- 形态学梯度: 膨胀 - 腐蚀,用于提取图像中的边缘
- 顶帽: 原图 - 开运算,用于提取图像中的小噪声
- 黑帽: 闭运算 - 原图,用于填充图像中的小孔洞