计算视频PSNR值代码

计算视频的PSNR值需要用到视频的原始帧和编码后的帧，以下是Python代码示例：

import cv2
import numpy as np

def calculate_psnr(original_frame, encoded_frame):
    # 读取原始帧和编码后的帧
    original = cv2.imread(original_frame)
    encoded = cv2.imread(encoded_frame)

    # 计算MSE
    mse = np.mean((original - encoded) ** 2)

    # 计算PSNR
    if mse == 0:
        psnr = 100
    else:
        max_pixel = 255.0
        psnr = 20 * np.log10(max_pixel / np.sqrt(mse))

    return psnr

其中，`original_frame`为原始帧的路径，`encoded_frame`为编码后的帧的路径。该函数返回视频的PSNR值。