Python OpenCV图像处理：使用Canny算子进行边缘检测

小蓝 • 2024-12-10 18:18 • 编程

一、介绍

Python OpenCV是一个非常强大的图像处理库，它可以让我们轻松地进行各种图片和视频的处理。其中，Canny算子是一种非常常用的边缘检测算法，它可以快速地找出图像中的边缘。本文将介绍如何使用Python OpenCV的Canny算子进行边缘检测。

二、Canny算子的基本原理

Canny算子是一种边缘检测算法，它是由John Canny在1986年提出的，是一种非常著名的图像处理算法。Canny算子的基本思想是：

使用高斯滤波器对图像进行平滑处理，去除噪声。
计算图像的梯度方向和大小。
使用非最大抑制（NMS）技术进行边缘细化，使边缘变得更加精细。
使用双阈值算法进行边缘检测，最终确定哪些边缘是真正的边缘。

三、使用Canny算子进行边缘检测的代码实现

import cv2

img = cv2.imread('test.jpg', 0)

# 高斯滤波器
img = cv2.GaussianBlur(img, (3, 3), 0)

# 计算梯度方向和大小
sobelx = cv2.Sobel(img, cv2.CV_64F, 1, 0, ksize=3)
sobely = cv2.Sobel(img, cv2.CV_64F, 0, 1, ksize=3)
grad = cv2.sqrt(sobelx ** 2 + sobely ** 2)
theta = cv2.phase(sobelx, sobely)

# 非最大抑制
grad_nms = cv2.copyMakeBorder(grad, 1, 1, 1, 1, cv2.BORDER_CONSTANT, 0)
for i in range(1, grad.shape[0]+1):
    for j in range(1, grad.shape[1]+1):
        if grad[i-1, j-1] == 0:
            grad_nms[i, j] = 0
        else:
            gradx = sobelx[i-1, j-1]
            grady = sobely[i-1, j-1]
            if np.abs(grady) > np.abs(gradx):
                weight = np.abs(gradx) / np.abs(grady)
                grad0 = grad_nms[i-1, j]
                grad1 = grad_nms[i+1, j]
                if gradx * grady > 0:
                    grad2 = grad_nms[i-1, j-1]
                    grad3 = grad_nms[i+1, j+1]
                else:
                    grad2 = grad_nms[i+1, j-1]
                    grad3 = grad_nms[i-1, j+1]
            else:
                weight = np.abs(grady) / np.abs(gradx)
                grad0 = grad_nms[i, j-1]
                grad1 = grad_nms[i, j+1]
                if gradx * grady > 0:
                    grad2 = grad_nms[i+1, j-1]
                    grad3 = grad_nms[i-1, j+1]
                else:
                    grad2 = grad_nms[i-1, j-1]
                    grad3 = grad_nms[i+1, j+1]
            grad_nms[i, j] = grad[i-1, j-1] if grad[i-1, j-1] >= weight * grad0 and grad[i-1, j-1] >= weight * grad1 and grad[i-1, j-1] >= weight * grad2 and grad[i-1, j-1] >= weight * grad3 else 0

# 双阈值算法
img_canny = np.zeros(grad_nms.shape, dtype=np.uint8)
thresh1 = 100
thresh2 = 200
for i in range(grad_nms.shape[0]):
    for j in range(grad_nms.shape[1]):
        if grad_nms[i, j]  thresh2:
            img_canny[i, j] = 255
        else:
            img_canny[i, j] = 50
            for a in range(-1, 2):
                for b in range(-1, 2):
                    if img_nms[i+a, j+b] > thresh2:
                        img_canny[i, j] = 255
                        break
                if img_canny[i, j] == 255:
                    break

cv2.imshow('Canny Edge Detection', img_canny)
cv2.waitKey(0)
cv2.destroyAllWindows()

四、Canny算子的优点

Canny算子是一种非常优秀的边缘检测算法，它具有以下几个优点：

准确性高：Canny算子可以找到一条边缘的精确位置。
抗噪声能力强：Canny算子在进行边缘检测之前进行了高斯平滑处理，可以去除图像中的噪声。
边缘细化效果好：Canny算子使用了非最大抑制技术和双阈值算法，可以使检测出的边缘更加精细。

五、总结

本文详细介绍了如何使用Python OpenCV的Canny算子进行边缘检测。首先介绍了Canny算子的基本原理，然后给出了使用Python OpenCV进行边缘检测的代码实现，并列举了Canny算子的优点。读者可以根据本文的内容自行编写代码进行尝试。

原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/230747.html