联合稀疏SOMP代码

来源:互联网 发布:淘宝账号和密码全忘记 编辑:程序博客网 时间:2024/05/28 23:10

原理网上有好多,在此就不写啦,这个是我自己在OMP基础上稍微改进的,作为一个函数可以调用,大家转载时要注明出处哟~~

function A=SOMP1(D,X,L) 
% 输入参数: 
%       D - 过完备字典,注意:必须字典的各列必须经过了规范化
%       X - 信号
%       L - 系数中非零元个数的最大值(可选,默认为D的列数,速度可能慢)
% 输出参数: 
%       A - 稀疏系数
%如果函数的输入参数只有两个时即D和X,则L默认为D的列数
P=size(X,2);%返回X的列数,如果是size(X,1)返回的是X的行数
K=size(D,2);
   temp=zeros(K,P);
    a=zeros(K,P);
    x=X;
    residual=x; %残差
    indx=zeros(L,1);%索引集
    for j=1:1:L
       proj=sum(residual'*D);
        [maxVal,pos]=max(abs(proj));
        pos=pos(1);
        indx(j)=pos;
         a=inv(D(:,indx(1:j))'*D(:,indx(1:j)))*D(:,indx(1:j))'*X;
         residual=x-D(:,indx(1:j))*a;
         temp(indx(1:j),:)=a;
          A=sparse(temp);
         if sum(sum(residual.^2)) < 1e-6
            break;
         end

    end

return;

1 0
原创粉丝点击