用Python计算两图像的峰值信噪比PSNR
图像信噪比是用于评估图像超分结果的指标
def psnr(img1, img2):
mse = np.mean( (img1/255. - img2/255.) ** 2 )
if mse < 1.0e-10:
return 100
PIXEL_MAX = 1
return 20 * math.log10(PIXEL_MAX / math.sqrt(mse))
1.首先计算mse。
对于三通道的RGB图像 计算mse的数学表达式是:
转换成代码来写:
2.得到mse后计算PSNR。
转换为代码来写: