1. 程式人生 > >用自編的程式實現小波影象分解與重構

用自編的程式實現小波影象分解與重構

function y=mywavedec2(x,dim)
% 函式 MYWAVEDEC2() 對輸入矩陣 x 進行 dim 層分解,得到相應的分解係數矩陣 y
% 輸入引數:x —— 輸入矩陣;
%           dim —— 分解層數。
% 輸出引數:y —— 分解係數矩陣。

x=modmat(x,dim);            % 首先規範化輸入矩陣,使其行列數均能被 2^dim 整除

subplot(121);imshow(x);title('原始影象');   % 畫出規範化後的源影象
[m,n]=size(x);              % 求出規範化矩陣x的行列數
xd=double(x);               % 將矩陣x的資料格式轉換為適合數值處理的double格式


for i=1:dim
    xd=modmat(xd,1);
    [dLL,dHL,dLH,dHH]=mydwt2(xd);   % 矩陣小波分解
    tmp=[dLL,dHL;dLH,dHH];          % 將分解係數存入快取矩陣
    xd=dLL;                         % 將快取矩陣左上角部分的子矩陣作為下一層分解的源矩陣
    [row,col]=size(tmp);            % 求出快取矩陣的行列數
    y(1:row,1:col)=tmp;             % 將快取矩陣存入輸出矩陣的相應行列
end

yd=uint8(y);            % 將輸出矩陣的資料格式轉換為適合顯示影象的uint8格式

for i=1:dim             % 對矩陣 yd 進行分界線處理,畫出分解影象的分界線
    m=m-mod(m,2);
    n=n-mod(n,2);
    yd(m/2,1:n)=255;
    yd(1:m,n/2)=255;
    m=m/2;n=n/2;
end
subplot(122);imshow(yd);title([ num2str(dim) ' 維小波分解影象']);