yuv转png的python代码

YUV（通常用于视频编码，其中Y代表亮度信息，U和V分别代表色度信息）转换成PNG图像文件是一个常见的计算机视觉任务。Python中有多种库可以完成这个任务，比如`opencv-python`。下面是一个简单的示例代码：

import cv2
import numpy as np

# 假设yuv_data是一个包含YUV数据的数组，通常是4D数组（height, width, 1, channels）
def yuv_to_png(yuv_data):
    # 将YUV数据从4D转换到3D（去除时间维度）
    yuv_3d = yuv_data.reshape(-1, yuv_data.shape[-1])

    # 分离Y、U、V通道
    y, u, v = yuv_3d[:, 0], yuv_3d[:, 1] * 0.5 + 128, yuv_3d[:, 2] * 0.5 + 128

    # 转换为RGB
    img_rgb = cv2.cvtColor(np.stack([y, u - 16, v - 128], axis=-1), cv2.COLOR_YUV2BGR_I420)

    # 将RGB图像保存为PNG
    cv2.imwrite('output.png', img_rgb)

# 使用示例
yuv_data = ... # 这里需要替换为实际的YUV数据
yuv_to_png(yuv_data)

注意：这个例子假设输入的是I420格式的YUV数据，如果是其他格式，可能需要调整分量的处理方式。此外，你需要安装`opencv-python`库才能运行这段代码。