psnr的python完整代码

以下是一个简单的 Python 代码，用于计算两个图像之间的 PSNR 值。

import math
import cv2

# 读入原图和处理后的图像
original = cv2.imread('original_image.jpg')
processed = cv2.imread('processed_image.jpg')

# 计算 MSE 值
mse = ((original - processed) ** 2).mean()

# 计算 PSNR 值
if mse == 0:
    psnr = float('inf')
else:
    max_pixel_value = 255.0
    psnr = 20 * math.log10(max_pixel_value / math.sqrt(mse))

# 输出结果
print('PSNR:', psnr)

需要注意的是，这个代码假设两个图像的分辨率相同，如果分辨率不同，需要先将两个图像的分辨率调整为一致。