Python计算两张图的PSNR值

Python计算两张图的PSNR值

    本文主要记录如何用python实现求解两张图的PSNR值。

1.PSNR求解公式

    （1）对于三通道的RGB图像 计算MSE的数学表达式是：

    具体的代码为：

# compute MSE
mse = np.mean((img1/1.0-img2/1.0)**2)

    （2）得到MSE后计算PSNR，具体数学表达式为：

    具体的代码为：

psnr1=20*math.log10(255/math.sqrt(mse))

    另外一种方式就是将像素值进行归一化处理，然后再进行PSNR值的求解：

mse = np.mean((img1/255.0-img2/255.0)**2)

psnr2=20*math.log10(1/math.sqrt(mse))

2.完整的PSNR求解代码

    完整的代码如下所示：