图像PSNR值及MSE值的计算

最新推荐文章于 2023-12-06 08:33:19 发布

最新推荐文章于 2023-12-06 08:33:19 发布 · 2.3k 阅读

文章标签：

#人工智能 #python

本文介绍了一种计算图像峰值信噪比(PSNR)和均方根误差(MSE)的方法，这两种指标常用于衡量图像质量。文章提供了一个MATLAB函数实现，该函数可以接受两个图像作为输入并返回它们的PSNR和MSE值。文中还特别提到，在使用此函数前需确保输入图像大小一致，并将图像转换为double类型。

% 图像峰值信噪比PSNR、均方根误差MSE
% By lyqmath
% Xidian University 
function [PSNR, MSE] = psnr(X, Y)
% 计算峰值信噪比PSNR、均方根误差MSE
% 如果输入Y为空，则视为X与其本身来计算PSNR、MSE

if nargin<2
    D = X;
else
    if any(size(X)~=size(Y))
        error('The input size is not equal to each other!');
    end
    D = X-Y;
end
MSE = sum(D(:).*D(:))/prod(size(X));
PSNR = 10*log10(255^2/MSE);

最后使用的时候注意把X,Y转换成double值...