基于小波变换的图像边缘检测（matlab祖传代码注释）

Brenda ·

更新时间:2024-11-13

· 528 次阅读

基于小波变换的图像边缘提取应用展示

图1 针对png格式无背景原图的边缘检测
上图为针对png格式无背景原图的边缘检测，对比各种边缘检测算子，小波变化的优势体现并不明显。

上图为针对含背景图片的边缘检测，小波变化的优势这里体现的比较明显。

matlab祖传代码注释

相信看到这篇文章的小伙伴一定已经搜索了“小波变换边缘检测”这样的关键词很多次了，但是得到的代码好像只有那一个，哈哈哈，我当时也是只发现了这一个代码，在查阅了一些文献后对代码思路有一些理解，在这里和大家分享~
边缘检测流程图
废话不多说，先放一张流程图便于大家理解代码。下面为加了一些注释的祖传代码。

clear all; 
I = imread('E:\DAT\postgraudate\class\信号\matlab\test\retest.jpg');    %读取照片
subplot(2,3,1);
imshow(I);
I = rgb2gray(I);subplot(2,3,2);imshow(I);   %彩色照片转灰度图 
I1 = imadjust(I,stretchlim(I),[0,1]);   %增强图片对比度，白的更白，黑的更黑
subplot(2,3,3);
imshow(I1);   
[N,M] = size(I);    %获取图片尺寸 
h = [0.125,0.375,0.375,0.125];     %尺度向量 相当于低通滤波器
g = [0.5,-0.5];     %相当于高通滤波器
delta = [1,0,0]; 
J = 2;     % 分解尺度
a(1:N,1:M,1,1:J+1) = 0;     %初始化变量 
dx(1:N,1:M,1,1:J+1) = 0; 		%x方向偏导初始化
dy(1:N,1:M,1,1:J+1) = 0;  		%y方向偏导初始化
d(1:N,1:M,1,1:J+1) = 0; 		% 模值初始化
a(:,:,1,1) = conv2(h,h,I,'same');   %卷积运算 
dx(:,:,1,1) = conv2(delta,g,I,'same'); 	%计算x方向偏导
dy(:,:,1,1) = conv2(g,delta,I,'same');  	%计算y方向偏导
x = dx(:,:,1,1); 
y = dy(:,:,1,1); 
d(:,:,1,1) = sqrt(x.^2+y.^2);   %计算模值
I1 = imadjust(d(:,:,1,1),stretchlim(d(:,:,1,1)),[0 1]);     %增强对比度
subplot(2,3,4);
imshow(I1); 
lh = length(h); 
lg = length(g); 
 for j = 1:J    %按照分解尺度进行分解
    lhj = 2^j*(lh-1)+1; 
    lgj = 2^j*(lg-1)+1; 
    hj(1:lhj)=0; 
    gj(1:lgj)=0; 
    for n = 1:lh 
        hj(2^j*(n-1)+1)=h(n); 
    end 
    for n = 1:lg 
        gj(2^j*(n-1)+1)=g(n); 
    end 
    a(:,:,1,j+1) = conv2(hj,hj,a(:,:,1,j),'same');    
    dx(:,:,1,j+1) = conv2(delta,gj,a(:,:,1,j),'same'); 
    dy(:,:,1,j+1) = conv2(gj,delta,a(:,:,1,j),'same'); 
    x = dx(:,:,1,j+1); 
    y = dy(:,:,1,j+1); 
    dj(:,:,1,j+1) = sqrt(x.^2+y.^2); 
    I1 = imadjust(dj(:,:,1,j+1),stretchlim(dj(:,:,1,j+1)),[0 1]);
    %subplot(2,3,4+j);
    figure();
    imshow(I1); 
 end
%以下部分为噪点滤波，和小波变化无关
Icomp = imcomplement(I1);   %反转图像
Ifilled = imfill(Icomp, 'holes');
se = strel('disk', 15);     %小于此半径的连通域忽略掉
Iopenned = imopen(Ifilled, se);
%subplot(2,4,4+j+1);
figure();
imshow(Iopenned);

噪点滤波作用如下图所示
在这里插入图片描述
大家看到这里可能对每一行代码大概知道时什么意思了，但是整个代码段的逻辑结构可能还不清楚。就比如说，我现在去买菜，我知道了现在是我在给他钱，下一步是他找我钱，但是，我为毛要去买菜啊！这是个问题。

代码思路

小波变换与尺度函数
时间多的，或者想融汇贯通这段代码的，点开上面这个链接，仔细看完，相信对你有很多帮助。我也是看了这篇文章对小波变换才知一二，没时间的，或者喜欢我的（哈哈哈）可以看我接下来的总结（扯淡）。

首先，小波变换分为连续小波变换和离散小波变换，由于我们用于计算的信号都是经过采样后的，所以接下来我们讲离散小波变化，同样代码里用的也是离散小波变换，这里可能有同学就要问了，为什么不用连续小波变换？是不是瞧不起我！

那么这里简单说一下，首先你怎么在代码里实现积分？如果有写过PID算法的同学可能知道，PID公式里也有积分，但是在代码里我们把他转变成了累加，这同时也是积分的意义，将坐标轴分成无限小的小段。。。扯远了

好的，现在我们确定使用离散小波变换了。
在这里插入图片描述
大家可以看到上图中信号输出 f(t) 等于后面的一串式子，那么我们怎么计算f(t)呢？

很简单，第一种方式，就按表达式算。那么我们是不是要有尺度函数和小波函数？那从哪得到呢？万能的淘宝也没有，需要自己构建，怎么构建呢？小波变换与尺度函数这个链接又来啦，详见里面所述。总之就是太麻烦了，所以代码里采用第二种方式。

第二种方式，设计低通滤波器h(k),利用鱼骨型算法迭代计算。这个 h(k) 是什么呢？
h = [0.125,0.375,0.375,0.125];
没错，就是代码里这个参数，代表尺度向量，至于参数初始值为什么是这个，则需要根据尺度函数需要满足的3个条件来进行计算。
有了h和g后我们利用下面这个公式就可以一层层进行计算小波函数系数和尺度函数系数。
在这里插入图片描述
同样这两个公式在代码里也有体现。分解尺度 j 也就是我们代码里的循环次数。
到这里大家可能知道一些代码为什么是这样写了，知道我们原来买菜是为了做饭，那又为什么要做饭呢？这个问题还得大家一起去探讨。这里放几个我在学习小波变换的边缘检测中遇到的好文章，共享。