waveletentropy.m

求解一维心电图的小波熵

%求小波包的信息熵，中南大学数学学院信科0302班张扬编写
%ECG为信号源，n为分解的层数，wpname为小波名字，我常用的是db3,haar,db4.
function y=waveletentropy(ECG,n,wpname)
wpt1=wpdec(ECG,n,wpname);        %对数据进行小波包分解
for i=1:2*n
      %wpcoef(wpt1,[n,i-1])是求第n层第i个节点的系数
      disp('每个节点的能量E(i)');
  E(i)=norm(wpcoef(wpt1,[n,i-1]),2)*norm(wpcoef(wpt1,[n,i-1]),2)
      %求第i个节点的范数平方，其实也就是平方和
end
 disp('小波包分解总能量E_total');
E_total=sum(E)  %求总能量
disp('以下是每个节点的概率P');
for i=1:2*n
   p(i)= E(i)/E_total    %求每个节点的概率
end
 %以下计算小波熵，即－sum（pj*lnpj），
 disp('小波熵的值S_wt');
for i=2:2*n
  m(i)=p(i)*log(p(i));
end
 S_wt=sum(m)*(-1)
 disp('小波包分解图形');
 text(1,1,'原始图象 概率分布图象 能量数值图像') ;
% subplot(1,3,1);
% plot(ECG);%原始图像
% subplot(1,3,2);
% bar(p); %概率分布图像
% subplot(1,3,3);
 bar(E); %能量数值图像