canny滤波器

Halcom

1. function canny_im = canny_fspecial(im,type)
   
2.     if nargin < 2
   
3.         type = 'canny';
   
4.     end
   
5.     if ~isa(im,'double')
   
6.         im = double(im)/255;
   
7.     end
   
8.    switch type
   
9.        case 'canny'  % canny filter
   
10. 
    
11.     r=im(:,:,1);g=im(:,:,2);b=im(:,:,3);
    
12.     % 平滑滤波器
    
13.     filter= [2 4 5 4 2;
    
14.              4 9 12 9 4;
    
15.              5 12 15 12 5;
    
16.              4 9 12 9 4;
    
17.              2 4 5 4 2];
    
18.     filter=filter/115;
    
19. 
    
20.     im_conv= convn(im,filter);  % 平滑滤波后图像
    
21. 
    
22.     % 计算梯度
    
23.     gradXfilt=[-1 0 1;   % 卷积模板convolution mask
    
24.                -2 0 2;
    
25.                -1 0 1];
    
26.     gradYfilt=[1  2   1; % 卷积模板 convolution mask
    
27.                0  0   0;
    
28.               -1  -2  -1];
    
29.     GradX= convn(im_conv,gradXfilt);  % 卷积
    
30.     GradY= convn(im_conv,gradYfilt);  % 卷积
    
31.     absgrad=abs(GradX)+abs(GradY);
    
32.    
    
33.     % 计算梯度角
    
34.     [a,b]=size(GradX);
    
35.     theta=zeros([a b]);
    
36.     for i=1:a
    
37.           for j=1:b
    
38.                 if(GradX(i,j)==0)
    
39.                    theta(i,j)=atan(GradY(i,j)/1e-10);
    
40.                 else
    
41.                     theta(i,j)=atan(GradY(i,j)/GradX(i,j));
    
42.                 end
    
43.           end
    
44.      end
    
45.       theta=theta*(180/3.14);
    
46.       for i=1:a
    
47.           for j=1:b
    
48.                 if(theta(i,j)<0)
    
49.                     theta(i,j)= theta(i,j)-90;
    
50.                     theta(i,j)=abs(theta(i,j));
    
51.                 end
    
52.           end
    
53.      end
    
54.       for i=1:a
    
55.           for j=1:b
    
56.               if ((0<theta(i,j))&&(theta(i,j)<22.5))||((157.5<theta(i,j))&&(theta(i,j)<181))
    
57.                     theta(i,j)=0;
    
58.               elseif (22.5<theta(i,j))&&(theta(i,j)<67.5)
    
59.                      theta(i,j)=45;
    
60.               elseif (67.5<theta(i,j))&&(theta(i,j)<112.5)  
    
61.                       theta(i,j)=90;
    
62.               elseif (112.5<theta(i,j))&&(theta(i,j)<157.5)
    
63.                       theta(i,j)=135;
    
64.               end
    
65.           end
    
66.       end
    
67. 
    
68.     % 非极大值抑制
    
69.     canny_im = padarray(absgrad, [1 1]);
    
70. % A = [1 2; 3 4];
    
71. % B = padarray(A,[3 2],'replicate','post')
    
72. % B =
    
73. %      1     2     2     2
    
74. %      3     4     4     4
    
75. %      3     4     4     4
    
76. %      3     4     4     4
    
77. %      3     4     4     4
    
78.     [a,b]=size(theta);
    
79.     for i=2:a-2
    
80.         for j=2:b-2
    
81.                if (theta(i,j)==135)
    
82.                      if ((canny_im(i-1,j+1)>canny_im(i,j))||(canny_im(i+1,j-1)>canny_im(i,j)))
    
83.                           canny_im(i,j)=0;
    
84.                       end
    
85.                elseif (theta(i,j)==45)   
    
86.                       if ((canny_im(i+1,j+1)>canny_im(i,j))||(canny_im(i-1,j-1)>canny_im(i,j)))
    
87.                            canny_im(i,j)=0;
    
88.                       end
    
89.                elseif (theta(i,j)==90)   
    
90.                       if ((canny_im(i,j+1)>canny_im(i,j))||(canny_im(i,j-1)>canny_im(i,j)))
    
91.                           canny_im(i,j)=0;
    
92.                       end
    
93.                elseif (theta(i,j)==0)   
    
94.                       if ((canny_im(i+1,j)>canny_im(i,j))||(canny_im(i-1,j)>canny_im(i,j)))
    
95.                           canny_im(i,j)=0;
    
96.                       end
    
97.                end
    
98.         end
    
99.     end
    
100. 
     
101.    end
     
102. end

复制代码