CS Primary tutorial CS压缩传感的初级教学代码

CS压缩传感的初级教学代码，使用OMP重构，已注释，包括1维信号，2维图像的重构，分别使用dct和小波稀疏，列扫描和分块法进行omp重构

%  1-D信号压缩传感的实现（正交匹配追踪法Orthogonal Matching Pursuit）
%  测量数M>=K*log（N/K）K是稀疏度N信号长度可以近乎完全重构
clc;clear

%%  1. 时域测试信号生成
K=7;      %  稀疏度（做FFT可以看出来）
N=256;    %  信号长度
M=64;     %  测量数（M>=K*log（N/K）至少40但有出错的概率）
f1=50;    %  信号频率1
f2=100;   %  信号频率2
f3=200;   %  信号频率3
f4=400;   %  信号频率4
fs=800;   %  采样频率
ts=1/fs;  %  采样间隔
Ts=1:N;   %  采样序列
x=0.3*cos（2*pi*f1*Ts*ts）+0.6*cos（2*pi*f2*Ts*ts）+0.1*cos（2*pi*f3*Ts*ts）+0.9*cos（2*pi*f4*Ts*ts）;  %  完整信号 1*256

%%  2.  时域信号压缩传感
Phi=randn（MN）;                                   %  测量矩阵（高斯分布白噪声） 64*256
s=Phi*x‘;                                        %  获得线性测量 64*1

%%  3.  正交匹配追踪法重构信号（本质上是L_1范数最优化问题）
m=2*K;                                            %  算法迭代次数（m>=K） 14
Psi=fft（eye（NN））/sqrt（N）                       %  傅里叶正变换矩阵 256*256
T=Phi*Psi‘;                                       %  恢复矩阵（测量矩阵*正交反变换矩阵） 64*256 传感矩阵

hat_y=zeros（1N）;                                 %  待重构的谱域（变换域）向量   1*N                  
Aug_t=[];                                         %  增量矩阵（初始值为空矩阵） 
r_n=s;                                            %  残差值 初始化为64*1

for times=1:m;                                    %  迭代次数（有噪声的情况下该迭代次数为K） 1:14
    for col=1:N;                                  %  恢复矩阵的所有列向量 1:256
        product（col）=abs（T（:col）‘*r_n）;          %  恢复矩阵的列向量和残差的投影系数（内积值）  
    end
    [valpos]=max（product）;                       %  最大投影系数对应的位置
     pos_array（times）=pos;                         %  更新索引集，即纪录最大投影系数的位置
    Aug_t=[Aug_tT（:pos）];                       %  矩阵扩充
    T（:pos）=zeros（M1）;                          %  选中的列置零（实质上应该去掉，为了简单我把它置零）
    aug_y=（Aug_t‘*Aug_t）^（-1）*Aug_t‘*s;           %  最小二乘使aug_y最小，aug_y为恢复信号
    r_n=s-Aug_t*aug_y;                            %  残差
   
end
hat_y（pos_array）=aug_y;                           %  重构的谱域向量
hat_x=real（Psi‘*hat_y.‘）;                         %  做逆傅里叶变换重构得到时域信号

%%  4.  恢复信号和原始信号对比
figure（1）;
hold on;
plot（hat_x‘k.-‘）                                 %  重建信号
plot（x‘r‘）                                       %  原始信号
legend（‘Recovery‘‘Original‘）
norm（hat_x.‘-x）/norm（x）                           %  重构误差

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----

     文件        874  2012-11-26 11:17  2D_image_CS_toturial.m

     文件       1358  2012-11-26 11:18  omp.m

     文件       2542  2012-11-19 18:33  1D_signal_CS_toturial.m

     文件       1236  2012-11-26 11:29  DWT.m

     文件       1329  2012-11-26 11:32  分块法重构\block_omp.m

     文件       1083  2012-11-20 23:14  分块法重构\OMPerr.m

     文件      66614  2009-06-08 18:00  lena256.bmp

     目录          0  2012-11-26 11:35  分块法重构

----------- ---------  ---------- -----  ----

                75036                    8