Python实现Canny边缘检测算法——附详细源代码

Python实现Canny边缘检测算法
100 篇文章 ¥99.90 ¥299.90
本文介绍如何使用Python和OpenCV实现Canny边缘检测算法,包括图像灰度化、高斯滤波、梯度计算、非极大值抑制、双阈值处理和边缘连接等步骤,附带详细源代码。

Python实现Canny边缘检测算法——附详细源代码

Canny边缘检测是一种广泛应用于图像处理领域的算法,其能够精确、高效地检测出图像中的边缘。本文将基于Python语言实现Canny边缘检测算法,并附上完整的源代码,让读者能够更好地理解和运用该算法。

  1. 算法原理

Canny边缘检测算法是由John F. Canny在1986年提出的一种多级别边缘检测算法,其核心思想是通过滤波器对图像进行平滑处理,然后使用梯度算子检测出图像中的边缘,最后通过非极大值抑制和双阈值处理确定边缘。

  1. 实现步骤

(1)图像灰度化

Canny边缘检测算法首先需要将彩色图像转换为灰度图像,本文采用opencv库的cv2.cvtColor()函数实现。

(2)高斯滤波

为了减少噪声对边缘的影响,需要对灰度图像进行平滑处理,本文采用了高斯滤波器,使用cv2.GaussianBlur()函数实现。

(3)求取图像梯度

通过计算图像中每个像素的梯度幅值和方向,来检测出图像的边缘。本文使用OpenCV中的cv2.Sobel()函数实现。

(4)非极大值抑制

为了精确检测出边缘,需要对梯度方向进行判断,只有在梯度方向与垂直或水平方向夹角小于45度时才算边缘,将其他方向的梯度值置为0。本文采用了cv2.phase()函数来计算梯度方向,并通过数学计算实现非极大值抑制。

(5)双阈值处理

在经过以上步骤后,得到的边缘仍存在一定程度的误检和漏检。因此需要通过双阈值处理来进一步确定真正的边缘。本文将阈值分别设置为50和1

Canny算子是一种常用的边缘检测算法,其Python实现如下: ```python import cv2 import numpy as np def canny(image, low_threshold, high_threshold): gray = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY) blur = cv2.GaussianBlur(gray, (5, 5), 0) canny = cv2.Canny(blur, low_threshold, high_threshold) return canny ``` 其中,`image`为输入的图像,`low_threshold`和`high_threshold`为双阈值滤波的低阈值和高阈值。函数首先将图像转换为灰度图像,然后进行高斯模糊,最后使用Canny算子进行边缘检测。 另外,如果需要使用PyTorch实现Canny算子,可以使用以下代码: ```python import torch import torch.nn.functional as F def canny_pytorch(image, low_threshold, high_threshold): gray = torch.mean(image, dim=0, keepdim=True) blur = F.conv2d(gray, torch.ones(1, 1, 5, 5) / 25, padding=2) dx = F.conv2d(blur, torch.tensor([[-1, 0, 1], [-2, 0, 2], [-1, 0, 1]], dtype=torch.float32).unsqueeze(0).unsqueeze(0), padding=1) dy = F.conv2d(blur, torch.tensor([[-1, -2, -1], [0, 0, 0], [1, 2, 1]], dtype=torch.float32).unsqueeze(0).unsqueeze(0), padding=1) gradient = torch.sqrt(dx ** 2 + dy ** 2) theta = torch.atan2(dy, dx) theta[theta < 0] += np.pi theta[theta >= np.pi * 7 / 8] = 0 theta[(theta >= np.pi * 1 / 8) & (theta < np.pi * 3 / 8)] = np.pi / 4 theta[(theta >= np.pi * 3 / 8) & (theta < np.pi * 5 / 8)] = np.pi / 2 theta[(theta >= np.pi * 5 / 8) & (theta < np.pi * 7 / 8)] = np.pi * 3 / 4 suppressed = gradient.clone() for i in range(1, suppressed.shape[2] - 1): for j in range(1, suppressed.shape[3] - 1): if theta[0, 0, i, j] == 0: if (gradient[0, 0, i, j] <= gradient[0, 0, i, j - 1]) or (gradient[0, 0, i, j] <= gradient[0, 0, i, j + 1]): suppressed[0, 0, i, j] = 0 elif theta[0, 0, i, j] == np.pi / 4: if (gradient[0, 0, i, j] <= gradient[0, 0, i - 1, j + 1]) or (gradient[0, 0, i, j] <= gradient[0, 0, i + 1, j - 1]): suppressed[0, 0, i, j] = 0 elif theta[0, 0, i, j] == np.pi / 2: if (gradient[0, 0, i, j] <= gradient[0, 0, i - 1, j]) or (gradient[0, 0, i, j] <= gradient[0, 0, i + 1, j]): suppressed[0, 0, i, j] = 0 elif theta[0, 0, i, j] == np.pi * 3 / 4: if (gradient[0, 0, i, j] <= gradient[0, 0, i - 1, j - 1]) or (gradient[0, 0, i, j] <= gradient[0, 0, i + 1, j + 1]): suppressed[0, 0, i, j] = 0 suppressed[gradient < low_threshold] = 0 suppressed[gradient > high_threshold] = 255 for i in range(1, suppressed.shape[2] - 1): for j in range(1, suppressed.shape[3] - 1): if suppressed[0, 0, i, j] == 255: if (suppressed[0, 0, i - 1, j] == 0) and (suppressed[0, 0, i + 1, j] == 0) and (suppressed[0, 0, i, j - 1] == 0) and (suppressed[0, 0, i, j + 1] == 0): suppressed[0, 0, i, j] = 0 return suppressed ``` 其中,`image`为输入的图像,`low_threshold`和`high_threshold`为双阈值滤波的低阈值和高阈值。函数首先将图像转换为灰度图像,然后进行高斯模糊,接着计算图像的梯度和方向,进行非极大值抑制和双阈值滤波,最后进行边缘连接。
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符  | 博主筛选后可见
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

编码实践

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值