公式法将yuv视频转换成rbg格式的代码

下面是Python代码，用于将YUV视频转换为RGB格式：

import numpy as np
import cv2

def yuv2rgb(yuv_frame):
    y = yuv_frame[:, :, 0]
    u = yuv_frame[:, :, 1]
    v = yuv_frame[:, :, 2]

    r = y + 1.13983 * (v-128)
    g = y - 0.39465 * (u-128) - 0.58060 * (v-128)
    b = y + 2.03211 * (u-128)

    rgb_frame = np.zeros(yuv_frame.shape)
    rgb_frame[:, :, 0] = b
    rgb_frame[:, :, 1] = g
    rgb_frame[:, :, 2] = r

    return rgb_frame.astype(np.uint8)

# 读取YUV视频帧
yuv_frame = cv2.imread('input.yuv', cv2.IMREAD_UNCHANGED)

# 转换为RGB格式
rgb_frame = yuv2rgb(yuv_frame)

# 保存转换后的RGB视频帧
cv2.imwrite('output.png', rgb_frame)

其中，`yuv_frame`是一个三维数组，包含YUV格式的视频帧数据。具体而言，第一维表示行数，第二维表示列数，第三维包含三个通道，分别对应Y、U和V分量。

`yuv2rgb`函数先将Y、U、V三个分量分别提取出来，然后使用公式将它们转换成对应的R、G、B分量。最后，将R、G、B分量组合成一个RGB视频帧并返回。