图像算法学习 - 计算图像均方差(MES),峰值信噪比(PSNR)

function [MES , PSNR] = FuncPSNR(SrcImg, ImgRef, PixelBit)
% 计算图像均方差(MES),峰值信噪比(PSNR)
% SrcImg-原始图像, ImgRef-处理后图像, PixelBit-图像深度 
% History  : 2021.xx.xx - 初次创建文件
%            2021.xx.xx - 实现对图像直方图的测试
% *************************************************************************
[height,width] = size(SrcImg);
imgSrc = double(SrcImg);
imgRef = double(ImgRef); 
MAX  = 2^PixelBit - 1;%图像有多少灰度级
MES  = sum(sum((imgSrc-imgRef).^2))/(height*width);%均方差
PSNR = 20*log10(MAX/sqrt(MES)); %峰值信噪比
end