Matlab实现小波变换

时间:2022-01-10 00:41:08

文章和代码以及样例图片等相关资源,已经归档至【Github仓库:​​digital-image-processing-matlab​​】或者公众号【AIShareLab】回复 数字图像处理 也可获取。

目的

  1. Haar、尺度和小波函数;
  2. 比较函数wavefast 和函数wavedec2 的执行时间;
  3. 小波的方向性和边缘检测。

步骤

Haar、尺度和小波函数

[Lo_D,Hi_D,Lo_R,Hi_R]=wfilters('haar')
%Haar 变换的分解和重构滤波器的长度为2
waveinfo('haar');
[phi,psi,xval]=wavefun('haar',10);
xaxis=zeros(size(xval));
subplot(121);plot(xval,phi,'k',xval,xaxis,'--k');
axis([0 1 -1.5 1.5]);axis square;
title('haar scaling function');
subplot(122);plot(xval,psi,'k',xval,xaxis,'--k');
axis([0 1 -1.5 1.5]);axis square;
title('haar wavelet function');

使用haar 滤波器的一个简单FWT

f=magic(4)%使用Haar 的单尺度小波变换
[c1,s1]=wavedec2(f,1,'haar')
[c2,s2]=wavedec2(f,2,'haar')

比较函数wavefast 和函数wavedec2 的执行时间

function [varargout]=wavefilter(wname,type)

%Check the input and output arguments
error (nargchk(1,2,nargin));
if (nargin==1&nargout~=4)|(nargin==2&nargout~=2)
error('Invalid number of output arguments');
end
if nargin == 1 & ~ischar(wname)
error('WNAME must be a string');
end

if nargin == 2 & ~ischar(type)
error('TYPE must be a string');
end

switch lower(wname)
case {'haar','db1'}
ld=[1 1]/sqrt(2); hd=[-1 1]/sqrt(2);
lr=ld; hr=-hd;
case 'db4'
ld=[-1.059740178499728e-002 3.288301166698295e-002 ...
3.084138183598697e-002 -1.870348117188811e-001 ...
-2.798376941698385e-002 6.308807679295904e-001 ...
7.14846575525415e-001 2.303778133088552e-001];
t=(0:7);
hd=ld; hd(end:-1:1)=cos(pi*t).*ld;
lr=ld; lr(end:-1:1)=ld;
hr=cos(pi*t).*ld;
case 'sym4'
ld=[-7.576571478927333e-002 -2.963552764599851e-002 ...
4.976186676320155e-001 8.03738758059161e-001 ...
2.978577956052774e-001 -9.921954357684722e-002 ...
-1.260396726203783e-002 3.222310060404270e-002];
t=(0:7);
hd=ld; hd(end:-1:1)=cos(pi*t).*ld;
lr=ld; lr(end:-1:1)=ld;
hr=cos(pi*t).*ld;
case 'bior6.8'
ld=[0 1.908831736481291e-003 -1.914286129088767e-003 ...
-1.699063986760234e-002 1.193456527972926e-002 ...
4.973490349094079e-002 -7.726317316720414e-002 ...
-9.405920349573646e-002 4.207962846098268e-001 ...
8.259229974584023e-001 4.207962846098268e-001 ...
-9.405920349573646e-002 -7.726317316720414e-002 ...
4.973490349094079e-002 1.193456527972926e-002 ...
-1.699063986760234e-002 -1.914286129088767e-003 ...
1.908831736481291e-003];
hd=[0 0 0 1.442628250562444e-002 -1.446750489679015e-002 ...
-7.872200106262882e-002 4.036797903033992e-002 ...
4.178491091502746e-001 -7.589077294536542e-002 ...
4.178491091502746e-001 4.036797903033992e-002 ...
-7.872200106262882e-002 -1.446750489679015e-002 ...
1.442628250562444e-002 0 0 0 0];
t=(0:17);
lr=cos(pi*(t+1)).*hd;
hr=cos(pi*t).*ld;
case 'jpep9.7'
ld=[0 0.02674875741080976 -0.01686411844287495 ...
-0.07822326652898785 0.266861184428732 ...
0.6029490182363579 0.266861184428732 ...
-0.07822326652898785 -0.01686411844287495 ...
0.02674875741080976];
hd=[0 -0.09127176311424948 0.05754352622849957 ...
0.5912717631142470 -1.115087052456994 ...
0.5912717631142470 0.05754352622849957 ...
-0.09127176311424948 0 0];
t=(0:9);
lr=cos(pi*(t+1)).*hd;
hr=cos(pi*t).*ld;
otherwise
error('Unrecognizable wavelet name (WNAME).');
end

%Output the requested filters
if(nargin == 1)
varargout(1:4)={ld,hd,lr,hr};
else
switch lower(type(1))
case 'd'
varargout={ld,hd};
case 'r'
varargout={lr,hr};
otherwise
error('Unrecognizable filter TYPE.')
end
end


function [c,s]=wavefast(x,n,varargin)

error(nargchk(3,4,nargin));
if nargin == 3
if ischar(varargin{1})
[lp,hp]=wavefilter(varargin{1},'d');
else
error('Missing wavelet name');
end
else
lp=varargin{1};hp=varargin{2};
end
f1=length(lp);
sx=size(x);
if (ndims(x)~=2)|(min(sx<2))|~isreal(x)|~isnumeric(x)
error('X must be a real, numeric matrix');
end

if (ndims(lp)~=2)|~isreal(lp)|~isnumeric(lp) ...
|(ndims(hp)~=2)|~isreal(hp)|~isnumeric(hp) ...
|(f1~=length(hp)) | rem(f1,2)~=0
error(['LP and HP must be even and equal length real ,'...'numeric filter vectors']);
end

if ~isreal(n)|~isnumeric(n)|(n<1)|(n>log2(max(sx)))
error(['N must be a real scalar between 1 and' ...'log2(max(size(x)))']);
end
c=[];s=sx;app=double(x);

%for each decomposition...
for i=1:n
[app,keep]=symextend(app,f1);

rows=symconv(app,hp,'row',f1,keep);
coefs=symconv(rows,hp,'col',f1,keep);
c=[coefs(:)' c];
s=[size(coefs);s];
coefs=symconv(rows,lp,'col',f1,keep);
c=[coefs(:)' c];
rows=symconv(app,lp,'row',f1,keep);
coefs=symconv(rows,hp,'col',f1,keep);
c=[coefs(:)' c];
app=symconv(rows,lp,'col',f1,keep);
end
c=[app(:)' c];
s=[size(app);s];

function [y,keep]=symextend(x,f1)
keep=floor((f1+size(x)-1)/2);
y=padarray(x,[(f1-1) (f1-1)],'symmetric','both');


function [ratio, maxdiff]=fwtcompare(f,n,wname)
tic;
[c1,s1]=wavedec2(f,n,wname);
reftime=toc;
tic;
[c2,s2]=wavefast(f,n,wname);
t2=toc;


%compare the results
ratio=t2/(reftime+eps);
maxdiff=abs(max(c1-c2));


function y=symconv(x,h,type,f1,keep)
if strcmp(type,'row')
y=conv2(x,h);
y=y(:,1:2:end);
y=y(:,f1/2+1:f1/2+keep(2));
else
y=conv2(x,h');
y=y(1:2:end,:);
y=y(f1/2+1:f1/2+keep(1),:);
end

%比较两种方法下的速度
clear all
clc
f=imread('D:\图像库\DIP3E_CH07_Original_Images\Fig0701.tif','tif');
[ratio,maxdifference]=fwtcompare(f,5,'db4')

小波的方向性和边缘检测

clear all
clc
f=imread('D:\图像库\DIP3E_CH07_Original_Images\Fig0701.tif');
imshow(f);
[c,s]=wavefast(f,1,'sym4');
figure;wave2gray(c,s,-6);
[nc,y]=wavecut('a',c,s);
figure;wave2gray(nc,s,-6);
edges=abs(waveback(nc,s,'sym4'));
figure;imshow(mat2gray(edges));
%子程序
function w=wave2gray(c,s,scale,border)
error(nargchk(2,4,nargin));
if(ndims(c)~=2)|(size(c,1)~=1)
error('c must be a row vector');
end
if(ndims(s)~=2)|~isreal(s)|~isnumeric(s)|(size(s,2)~=2)
error('s must be a real,numeric two-column array');
end
elements=prod(s,2);

if (length(c)<elements(end))|...
~(elements(1)+3*sum(elements(2:end-1))>=elements(end))
error(['[c s] must be a standard wavelet' ...'decomposition structure']);
end
if(nargin>2)&(~isreal(scale)|~isnumeric(scale))
error('scale must be a real,numeric scalar');
end

if(nargin>3)&(~ischar(border))
error('border must be character string');
end

if nargin==2
scale=1;%default scale
end
if nargin<4
border='absord';%default border
end

absflag=scale<0;
scale=abs(scale);
if scale==0
scale=1;
end

[cd,w]=wavecut('a',c,s);
w=mat2gray(w);
cdx=max(abs(cd(:)))/scale;
if absflag
cd=mat2gray(abs(cd),[0,cdx]);
fill=0;
else
cd=mat2gray(cd,[-cdx,cdx]);
fill=0.5;
end
%build gray image one decomposition at a time
for i=size(s,1)-2:-1:1
ws=size(w);
h=wavecopy('h',cd,s,i);
pad=ws-size(h);
frontporch=round(pad/2);
h=padarray(h,frontporch,fill,'pre');
h=padarray(h,pad-frontporch,fill,'post');
v=wavecopy('v',cd,s,i);
pad=ws-size(v);
frontporch=round(pad/2);
v=padarray(v,frontporch,fill,'pre');
h=padarray(v,pad-frontporch,fill,'post');
d=wavecopy('d',cd,s,i);
pad=ws-size(d);
frontporch=round(pad/2);
d=padarray(d,frontporch,fill,'pre');
d=padarray(d,pad-frontporch,fill,'post');

%Add 1 pixel white border
switch lower(border)
case 'append'
w=padarray(w,[1 1],1,'post');
h=padarray(h,[1 0],1,'post');
v=padarray(v,[0 1],1,'post');
case 'absord'
w(:,end)=1;
w(end,:)=1;
h(end,:)=1;
v(:,end)=1;
otherwise
error('Unrecognized border parameter');
end
w=[w h;v d];
end

if nargout==0
imshow(w);%display result
end


function [varargout]=waveback(c,s,varargin)

error(nargchk(3,5,nargin));
error(nargchk(1,2,nargout));

if (ndims(c)~=2)|(size(c,1)~=1)
error('c must be a row vector');
end
if(ndims(s)~=2)|~isreal(s)|~isnumeric(s)|(size(s,2)~=2)
error('s must be a real,numeric two-column array');
end
elements=prod(s,2);

if (length(c)<elements(end))|...
~(elements(1)+3*sum(elements(2:end-1))>=elements(end))
error(['[c s] must be a standard wavelet' ...
'decomposition structure']);
end

nmax=size(s,1)-2;
wname=varargin{1};
filterchk=0;
nchk=0;

switch nargin
case 3
if ischar(wname)
[lp,hp]=wavefilter(wname,'r');
n=nmax;
else
error('updefined filter');
end
if nargout~=1
error('wrong number of output arguments');
end
case 4
if ischar(wname)
[lp,hp]=wavefilter(wname,'r');
n=varargin{2};
nchk=1;
else
lp=varargin{1};
hp=varargin{2};
filterchk=1;
n=nmax;
if nargout~=1
error('wrong number of output arguments');
end
end
case 5
lp=varargin{1};
hp=varargin{2};
filterchk=1;
n=varargin{3};
nchk=1;
otherwise
error('improper number fo input arguments');
end

f1=length(lp);
if filterchk
if (ndims(lp)~=2)|~isreal(lp)|~isnumeric(lp) ...
|(ndims(hp)~=2)|~isreal(hp)|~isnumeric(hp) ...
|(f1~=length(hp)) | rem(f1,2)~=0
error(['LP and HP must be even and equal length real ,'...
'numeric filter vectors']);
end
end

if nchk & (~isnumeric(n)|~isreal(n))
error('n must be a real numeric');
end
if (n>nmax)|(n<1)
error('Invalid number (N) of reconstructions requested');
end
if (n~=nmax) & (nargout~=2)
error('Not enough output arguments');
end

nc=c;
ns=s;
nnmax=nmax;
for i=1:n
a=symconvup(wavecopy('a',nc,ns),lp,lp,f1,ns(3,:))+ ...
symconvup(wavecopy('h',nc,ns,nnmax),hp,lp,f1,ns(3,:))+ ...
symconvup(wavecopy('v',nc,ns,nnmax),lp,hp,f1,ns(3,:))+ ...
symconvup(wavecopy('d',nc,ns,nnmax),hp,hp,f1,ns(3,:));

nc=nc(4*prod(ns(1,:))+1:end);
nc=[a(:)' nc];
ns=ns(3:end,:);
ns=[ns(1,:);ns];
nnmax=size(ns,1)-2;
end

if nargout==1;
a=nc;
nc=repmat(0,ns(1,:));
nc(:)=a;
end
varargout{1}=nc;
if nargout==2
varargout{2}=ns;
end

function z=symconvup(x,f1,f2,fln,keep)

y=zeros([2 1].*size(x));
y(1:2:end,:)=x;
y=conv2(y,f1');
z=zeros([1 2].*size(y));
z(:,1:2:end)=y;
z=conv2(z,f2);
z=z(fln-1:fln+keep(1)-2,fln-1:fln+keep(2)-2);

function [nc,y]=wavecut(type,c,s,n)
error(nargchk(3,4,nargin));
if nargin==4
[nc,y]=wavework('cut',type,c,s,n);
else
[nc,y]=wavework('cut',type,c,s);
End

function y=wavecopy(type,c,s,n)
error(nargchk(3,4,nargin));
if nargin==4
[nc,y]=wavework('cut',type,c,s,n);
else
[nc,y]=wavework('cut',type,c,s);
End

function y=wavepaste(type,c,s,n)
error(nargchk(5,5,nargin))
nc=wavework('paste',type,c,s,n,x);

function [varargout]=wavework(opcode,type,c,s,n,x)
error(nargchk(4,6,nargin));
if (ndims(c)~=2)|(size(c,1)~=1)
error('c must be a row vector');
end
if(ndims(s)~=2)|~isreal(s)|~isnumeric(s)|(size(s,2)~=2)
error('s must be a real,numeric two-column array');
end

elements=prod(s,2);

if (length(c)<elements(end))|...
~(elements(1)+3*sum(elements(2:end-1))>=elements(end))
error(['[c s] must be a standard wavelet' ...
'decomposition structure']);
end

if strcmp (lower(opcode(1:3)),'pas') & nargin < 6
error('Not enough input arguments');
end

if nargin<5
n=1;
end


nmax=size(s,1)-2;


aflag=(lower(type(1))=='a');
if ~aflag & (n>nmax)
error('N exceeds the decompositon in [c,s]');
end

switch lower(type(1))
case 'a'
nindex=1;
start=1;
stop=elements(1);
ntst=nmax;
case {'h','v','d'}
switch type
case 'h'
offset=0;
case 'v'
offset=1;
case 'd'
offset=2;
end
nindex=size(s,1)-n;
start=elements(1)+3*sum(elements(2:nmax-n+1))+ ...
offset*elements(nindex)-1;
stop=start+elements(nindex)-1;
ntst=n;
otherwise
error('Type must begin with "a","h","v",or "d"');
end

switch lower(opcode)
case {'copy','cut'}
y=repmat(0,s(nindex,:));
y(:)=c(start:stop);
nc=c;
if strcmp(lower(opcode(1:3)),'cut')
nc(start:stop)=0;
varargout={nc,y};
else
varargout={y};
end
case 'paste'
if prod(size(x))~=elements(end-ntst)
error('x is not sized for the requested paste');
else
nc=c;
nc(start:stop)=x(:);
varargout={nc};
end
otherwise
error('unrecognized OPCODE');
end

参考文献:

[1] ​​Rafael C. Gonzalez, Richard E. Woods, and Steven L. Eddins. 2003. Digital Image Processing Using MATLAB. Prentice-Hall, Inc., USA.​

[2] ​​阮秋琦. 数字图像处理(MATLAB版)[M]. 北京:电子工业出版社, 2014.​

[3] ​​冈萨雷斯. 数字图像处理(第三版)[M]. 北京:电子工业出版社, 2011.​