psnr.m

一个用matlab写的对图象处理方面的函数

function mySNR = psnr(img_A, img_B)
%psnr.m
[hight,width,dimension] = size(img_A);
sub_img = imsubtract( img_A, img_B);
sub_img = double(sub_img);
sub_img = sub_img .* sub_img;%do square

%square_img = sub_img .* sub_img;
Total = 0;
for j = 1 : hight        
    for i = 1 : width
    Total = Total + sub_img(j,i);        
%    Total = Total + square_img(j, i );    
    end
end

Pixels = hight * width;
Total = Total / Pixels;
up = 255 * 255;
mySNR = 99.99; 
if Total > 0
    SNR = up / Total;
    mySNR = 10 * log10(SNR);
end