matlab - 彩色 jpeg 图像的功率信噪比 (PSNR)

Question

我有一个原始图像和原始图像的扭曲图像。我想计算失真图像的 PSNR，以便以 dB 为单位测量失真。图像类型为彩色 jpeg。

Answer 1

我不知道您以前使用过什么，但您可以使用以下代码来计算更改后的图像的 PSNR：

I = imread('original.jpg');
Ihat = imread('changed.jpg');

% Read the dimensions of the image.
[rows columns ~] = size(I);

% Calculate mean square error of R, G, B.   
mseRImage = (double(I(:,:,1)) - double(Ihat(:,:,1))) .^ 2;
mseGImage = (double(I(:,:,2)) - double(Ihat(:,:,2))) .^ 2;
mseBImage = (double(I(:,:,3)) - double(Ihat(:,:,3))) .^ 2;

mseR = sum(sum(mseRImage)) / (rows * columns);
mseG = sum(sum(mseGImage)) / (rows * columns);
mseB = sum(sum(mseBImage)) / (rows * columns);

% Average mean square error of R, G, B.
mse = (mseR + mseG + mseB)/3;

% Calculate PSNR (Peak Signal to noise ratio).
PSNR_Value = 10 * log10( 255^2 / mse);

Answer 2

这是一个矢量化的实现：

mse = mean(mean((im2double(I) - im2double(K)).^2, 1), 2);
psnr = 10 * log10(1 ./ mean(mse,3));

它应该适用于整数和浮点图像，包括灰度和彩色图像。

我正在使用以下PSNR定义：