python中calculate_psnr函数

在Python中，`calculate_psnr()`通常是指计算 Peak Signal-to-Noise Ratio (PSNR) 的函数。PSNR是一种用于评估图像质量损失的指标，它衡量了原始图像与处理后的图像之间的差异程度。PSNR值越高，表明图像质量越好。

这个函数一般会接受两个参数，一个是参考图像（通常是未经压缩或噪声较小的原图），另一个是处理后的图像。函数内部会做像素级别的对比，计算每个像素点的均方误差（MSE），然后取其平均并求对数以得到dB值，公式大致如下：

def calculate_psnr(image1, image2):
    mse = np.mean((image1.astype('float') - image2.astype('float')) ** 2)
    if mse == 0:
        return float('inf')
    max_pixel_value = 255.0  # 或者根据图像深度调整
    psnr = 10 * np.log10(max_pixel_value**2 / mse)
    return psnr

在这个函数中，`np`代表NumPy库，`astype('float')`是为了将像素转换为浮点数以便进行精确的计算，`log10()`用于计算对数。注意，如果 MSE 等于零，这表示两张图片完全一致，此时 PSNR 被设置为无穷大。