边缘检测系列1:传统边缘检测算子
时间:2025-07-18 | 作者: | 阅读:0本文介绍了图像边缘检测的原理,指出边缘是灰度剧变处,检测基于方向导数掩码卷积。实现了通用边缘检测算子EdgeOP,集成Roberts、Prewitt等多种算子,提供四种边缘强度计算方式,并通过测试函数对比效果,展示了不同算子和计算方法的检测结果。
引入
图像的边缘指的是灰度值发生急剧变化的位置。
在图像形成过程中,由于亮度、纹理、颜色、阴影等物理因素的不同而导致图像灰度值发生突变,从而形成边缘。
边缘是通过检查每个像素的邻域并对其灰度变化进行量化的,这种灰度变化的量化相当于微积分里连续函数中方向导数或者离散数列的差分。
算法原理
传统的边缘检测大多数是通过基于方向导数掩码(梯度方向导数)求卷积的方法。
计算灰度变化的卷积算子包含Roberts算子、Prewitt算子、Sobel算子、Scharr算子、Kirsch算子、Robinson算子、Laplacian算子。
大多数边缘检测算子是基于方向差分卷积核求卷积的方法,在使用由两个或者多个卷积核组成的边缘检测算子时假设有 n 个卷积核,记?Conv1,Conv2,...,Convn,为图像分别与个卷积核做卷积的结果,通常有四种方式来衡量最后输出的边缘强度。
取对应位置绝对值的和:∑i=1n∣convi∣
取对应位置平方和的开方:∑i=1nconvi2
取对应位置绝对值的最大值:max{∣conv1∣,∣conv2∣,...,∣convi∣}
插值法:∑i=1nai∣convi∣,其中?ai>=0,且?∑i=1nai=1
代码实现
构建通用的边缘检测算子
- 因为上述的这些算子在本质上都是通过卷积计算实现的,只是所使用到的卷积核参数有所不同
- 所以可以构建一个通用的计算算子,只需要传入对应的卷积核参数即可实现不同的边缘检测
- 并且在后处理时集成了上述的四种计算最终边缘强度的方式
import numpy as npimport paddleimport paddle.nn as nnclass EdgeOP(nn.Layer): def __init__(self, kernel): ''' kernel: shape(out_channels, in_channels, h, w) ''' super(EdgeOP, self).__init__() out_channels, in_channels, h, w = kernel.shape self.filter = nn.Conv2D(in_channels=in_channels, out_channels=out_channels, kernel_size=(h, w), padding='SAME', bias_attr=False) self.filter.weight.set_value(kernel.astype('float32')) @staticmethod def postprocess(outputs, mode=0, weight=None): ''' Input: NCHW Output: NHW(mode==1-3) or NCHW(mode==4) Params: mode: switch output mode(0-4) weight: weight when mode==3 ''' if mode==0: results = paddle.sum(paddle.abs(outputs), axis=1) elif mode==1: results = paddle.sqrt(paddle.sum(paddle.pow(outputs, 2), axis=1)) elif mode==2: results = paddle.max(paddle.abs(outputs), axis=1) elif mode==3: if weight is None: C = outputs.shape[1] weight = paddle.to_tensor([1/C] * C, dtype='float32') else: weight = paddle.to_tensor(weight, dtype='float32') results = paddle.einsum('nchw, c -> nhw', paddle.abs(outputs), weight) elif mode==4: results = paddle.abs(outputs) return paddle.clip(results, 0, 255).cast('uint8') @paddle.no_grad() def forward(self, images, mode=0, weight=None): outputs = self.filter(images) return self.postprocess(outputs, mode, weight)登录后复制
图像边缘检测测试函数
- 为了方便测试就构建了如下的测试函数,测试同一张图片不同算子/不同边缘强度计算方法的边缘检测效果
import osimport cv2from PIL import Imagedef test_edge_det(kernel, img_path='test.jpg'): img = cv2.imread(img_path, 0) img_tensor = paddle.to_tensor(img, dtype='float32')[None, None, ...] op = EdgeOP(kernel) all_results = [] for mode in range(4): results = op(img_tensor, mode=mode) all_results.append(results.numpy()[0]) results = op(img_tensor, mode=4) for result in results.numpy()[0]: all_results.append(result) return all_results, np.concatenate(all_results, 1)登录后复制
Roberts 算子
roberts_kernel = np.array([ [[ [1, 0], [0, -1] ]], [[ [0, -1], [1, 0] ]]])_, concat_res = test_edge_det(roberts_kernel)Image.fromarray(concat_res)登录后复制
<PIL.Image.Image image mode=L size=3600x600 at 0x7F2548799F10>登录后复制
Prewitt 算子
In [4]prewitt_kernel = np.array([ [[ [-1, -1, -1], [ 0, 0, 0], [ 1, 1, 1] ]], [[ [-1, 0, 1], [-1, 0, 1], [-1, 0, 1] ]], [[ [ 0, 1, 1], [-1, 0, 1], [-1, -1, 0] ]], [[ [ -1, -1, 0], [ -1, 0, 1], [ 0, 1, 1] ]]])_, concat_res = test_edge_det(prewitt_kernel)Image.fromarray(concat_res)登录后复制
<PIL.Image.Image image mode=L size=4800x600 at 0x7F251781EF10>登录后复制
Sobel 算子
In [5]sobel_kernel = np.array([ [[ [-1, -2, -1], [ 0, 0, 0], [ 1, 2, 1] ]], [[ [-1, 0, 1], [-2, 0, 2], [-1, 0, 1] ]], [[ [ 0, 1, 2], [-1, 0, 1], [-2, -1, 0] ]], [[ [ -2, -1, 0], [ -1, 0, 1], [ 0, 1, 2] ]]])_, concat_res = test_edge_det(sobel_kernel)Image.fromarray(concat_res)登录后复制
<PIL.Image.Image image mode=L size=4800x600 at 0x7F251782E8D0>登录后复制
Scharr 算子
In [6]scharr_kernel = np.array([ [[ [-3, -10, -3], [ 0, 0, 0], [ 3, 10, 3] ]], [[ [-3, 0, 3], [-10, 0, 10], [-3, 0, 3] ]], [[ [ 0, 3, 10], [-3, 0, 3], [-10, -3, 0] ]], [[ [ -10, -3, 0], [ -3, 0, 3], [ 0, 3, 10] ]]])_, concat_res = test_edge_det(scharr_kernel)Image.fromarray(concat_res)登录后复制
<PIL.Image.Image image mode=L size=4800x600 at 0x7F251782EE10>登录后复制
Krisch 算子
In [7]Krisch_kernel = np.array([ [[ [5, 5, 5], [-3,0,-3], [-3,-3,-3] ]], [[ [-3, 5,5], [-3,0,5], [-3,-3,-3] ]], [[ [-3,-3,5], [-3,0,5], [-3,-3,5] ]], [[ [-3,-3,-3], [-3,0,5], [-3,5,5] ]], [[ [-3, -3, -3], [-3,0,-3], [5,5,5] ]], [[ [-3, -3, -3], [5,0,-3], [5,5,-3] ]], [[ [5, -3, -3], [5,0,-3], [5,-3,-3] ]], [[ [5, 5, -3], [5,0,-3], [-3,-3,-3] ]],])_, concat_res = test_edge_det(Krisch_kernel)Image.fromarray(concat_res)登录后复制
<PIL.Image.Image image mode=L size=7200x600 at 0x7F24FF09E4D0>登录后复制
Robinson算子
In [8]robinson_kernel = np.array([ [[ [1, 2, 1], [0, 0, 0], [-1, -2, -1] ]], [[ [0, 1, 2], [-1, 0, 1], [-2, -1, 0] ]], [[ [-1, 0, 1], [-2, 0, 2], [-1, 0, 1] ]], [[ [-2, -1, 0], [-1, 0, 1], [0, 1, 2] ]], [[ [-1, -2, -1], [0, 0, 0], [1, 2, 1] ]], [[ [0, -1, -2], [1, 0, -1], [2, 1, 0] ]], [[ [1, 0, -1], [2, 0, -2], [1, 0, -1] ]], [[ [2, 1, 0], [1, 0, -1], [0, -1, -2] ]],])_, concat_res = test_edge_det(robinson_kernel)Image.fromarray(concat_res)登录后复制
<PIL.Image.Image image mode=L size=7200x600 at 0x7F251782EED0>登录后复制
Laplacian 算子
In [9]laplacian_kernel = np.array([ [[ [1, 1, 1], [1, -8, 1], [1, 1, 1] ]], [[ [0, 1, 0], [1, -4, 1], [0, 1, 0] ]]])_, concat_res = test_edge_det(laplacian_kernel)Image.fromarray(concat_res)登录后复制
<PIL.Image.Image image mode=L size=3600x600 at 0x7F24FF0A6A90>登录后复制
总结
- 简单介绍并实现了几种常用的传统边缘检测算子
福利游戏
相关文章
更多-
- 大奉打更人日常任务怎么完成 帮会副本任务全解析
- 时间:2025-07-18
-
- 明日方舟界园肉鸽三结局boss怎么打 界园肉鸽三结局通关攻略
- 时间:2025-07-18
-
- 漫威争锋steam叫什么 漫威争锋英文名详细介绍
- 时间:2025-07-18
-
- 星痕共鸣巨塔遗迹副本怎么打 巨塔遗迹卡特格里夫打法分享
- 时间:2025-07-18
-
- 女子网购泳衣走光后被告知不能游泳 商家:海边度假泳衣
- 时间:2025-07-18
-
- 【Pipeline】林业病虫害防治AI昆虫识别项目
- 时间:2025-07-18
-
- 【PaddleDetection】结合51单片机搭建微信跳一跳物理外挂
- 时间:2025-07-18
-
- 『抽丝剥茧』深度解析PaddleClas—分类,一篇带你学会分类实践
- 时间:2025-07-18
大家都在玩
热门话题
大家都在看
更多-
- 女童患重病想退年卡 景区仅退款:年卡给小朋友留着
- 时间:2025-07-17
-
- 国外化妆品包装标注英寸容量,怎么换算成厘米对应体积,英寸和厘米怎么换算?
- 时间:2025-07-17
-
- 雅迪九号全网下架电动自行车 经销商:新国标将实施 正在“清库存”
- 时间:2025-07-17
-
- Uniswap如何购买
- 时间:2025-07-17
-
- 毒蘑菇病毒测试入口 点击进入毒蘑菇病毒性能测试
- 时间:2025-07-17
-
- 搜书吧官网入口2025 搜书吧官网最新2025进入
- 时间:2025-07-17
-
- BtcV暴跌!原因分析及抄底时机?
- 时间:2025-07-17
-
- 芬兰北极圈气温超30℃ 当地人热到光膀子上街 居民称“和南欧差不多一样了
- 时间:2025-07-17