视频质量评估指标PSNR的代码

可以使用以下Python代码计算图像的PSNR值：

import cv2
import numpy as np

def psnr(img1, img2):
    mse = np.mean((img1 - img2)**2)
    if mse == 0:
        return float('inf')
    max_pixel = 255.0
    psnr = 20 * np.log10(max_pixel / np.sqrt(mse))
    return psnr

# 用法示例
img1 = cv2.imread("img1.png")
img2 = cv2.imread("img2.png")
print(psnr(img1, img2))

其中 `img1` 和 `img2` 为两张需要比较的图片。函数返回值为它们之间的PSNR值。