matlab练习程序(SSIM)

  一种评价压缩后图像质量的方法,和PSNR类似。具体论文看这里。

  代码:

function main

    img=imread('lena.jpg');
    [m n]=size(img);
    imgn=imresize(img,[floor(m/8) floor(n/8)]);
    imgn=imresize(imgn,[m n]);
    img=double(img);
    imgn=double(imgn);

    tmp=[];
    for i=9:8:m
       for j=9:8:n
          tmp=[tmp SSIM(img(i-8:i,j-8:j),imgn(i-8:i,j-8:j))];        
       end    
    end

    MSSIM=mean(tmp)

    function re=SSIM(X,Y)  %返回值在0-1之间,越大质量越好。
        X=double(X);
        Y=double(Y);

        ux=mean(mean(X));
        uy=mean(mean(Y));

        sigma2x=mean(mean((X-ux).^2));
        sigma2y=mean(mean((Y-uy).^2));   
        sigmaxy=mean(mean((X-ux).*(Y-uy)));

        k1=0.01;
        k2=0.03;
        L=255;
        c1=(k1*L)^2;
        c2=(k2*L)^2;
        c3=c2/2;

        l=(2*ux*uy+c1)/(ux*ux+uy*uy+c1);
        c=(2*sqrt(sigma2x)*sqrt(sigma2y)+c2)/(sigma2x+sigma2y+c2);
        s=(sigmaxy+c3)/(sqrt(sigma2x)*sqrt(sigma2y)+c3);

        re=l*c*s;

    end

end

 

你可能感兴趣的:(matlab练习程序(SSIM))