峰值信噪比python

峰值信噪比（Peak Signal-to-Noise Ratio，PSNR）是一种用于衡量图像或视频量的指标，通常用于评估压缩算法的性能。在Python中，可以使用以下代码计算峰值信噪比：

import cv2
import numpy as np

def calculate_psnr(image1, image2):
    # 读取图片
    img1 = cv2.imread(image1)
    img2 = cv2.imread(image2)

    # 将图片转换为灰度图像
    gray_img1 = cv2.cvtColor(img1, cv2.COLOR_BGR2GRAY)
    gray_img2 = cv2.cvtColor(img2, cv2.COLOR_BGR2GRAY)

    # 计算均方误差
    mse = np.mean((gray_img1 - gray_img2) ** 2)
    
    # 计算最大像素值
    max_pixel = 255.0

    # 计算峰值信噪比
    psnr = 20 * np.log10(max_pixel / np.sqrt(mse))
    
    return psnr

# 示例用法
image1 = 'image1.jpg'
image2 = 'image2.jpg'
psnr_value = calculate_psnr(image1, image2)
print("PSNR:", psnr_value)

请注意，上述代码使用了OpenCV库，因此需要安装OpenCV才能运行。你需要将`image1.jpg`和`image2.jpg`替换为你要比较的两个图像的文件路径。函数`calculate_psnr`返回计算得到的PSNR值，单位为dB。