峰值信噪比python实现代码

峰值信噪比（Peak Signal-to-Noise Ratio, PSNR）是衡量图像质量的一个指标，它通常用于评估重构图像与原始图像之间的相似度。在Python中，我们可以使用OpenCV库来计算PSNR。这里是一个简单的例子：

import cv2
import numpy as np

def calculate_psnr(image1, image2):
    # 首先，确保两个输入图像具有相同的尺寸
    if image1.shape != image2.shape:
        raise ValueError("Images must have the same dimensions.")

    # 将图像转换为灰度（如果它们不是）
    if len(image1.shape) == 3 and image1.shape[-1] > 1:
        image1 = cv2.cvtColor(image1, cv2.COLOR_BGR2GRAY)
        image2 = cv2.cvtColor(image2, cv2.COLOR_BGR2GRAY)

    # 计算均值
    mean1 = np.mean(image1)
    mean2 = np.mean(image2)

    # 计算方差
    var1 = np.var(image1)
    var2 = np.var(image2)

    # 使用下面公式计算PSNR
    psnr = 20 * np.log10(np.max(image1)) - 10 * np.log10(var1 / (var2 + mean1**2))

    return psnr

# 使用示例
image1_path = 'path_to_your_image1.jpg'
image2_path = 'path_to_your_image2.jpg'

image1 = cv2.imread(image1_path)
image2 = cv2.imread(image2_path)

psnr_value = calculate_psnr(image1, image2)
print(f"PSNR value: {psnr_value} dB")