### YUV 转 JPG 的 Python 实现 要将 YUV 格式的图像转换为 JPG 格式，可以利用 `numpy` 和 `opencv-python` 库来完成此操作。以下是具体的方法： #### 方法概述 YUV 是一种颜色编码方式，通常用于视频压缩领域。为了将其转换为 JPG 图像文件，需要先加载原始数据并解析其像素格式，再通过 OpenCV 将其转换为 BGR 或 RGB 颜色空间。 以下是一个完整的代码示例[^1]： ```python import cv2 import numpy as np def yuv_to_jpg(yuv_file_path, width, height, output_path): # 计算帧大小 frame_size = width * height # 打开 YUV 文件 with open(yuv_file_path, 'rb') as f: # 读取 Y 数据 (亮度分量) y_data = np.frombuffer(f.read(frame_size), dtype=np.uint8).reshape((height, width)) # 读取 U 数据 (Cb 分量)，注意它是四分之一尺寸 u_data = np.frombuffer(f.read(frame_size // 4), dtype=np.uint8).reshape((height // 2, width // 2)) # 读取 V 数据 (Cr 分量)，同样也是四分之一尺寸 v_data = np.frombuffer(f.read(frame_size // 4), dtype=np.uint8).reshape((height // 2, width // 2)) # 创建空的 YUV 数组 yuv_image = np.empty((height * 3 // 2, width), dtype=np.uint8) # 填充 Y 组件到数组前半部分 yuv_image[0:height, :] = y_data # 插入 U 和 V 到数组后半部分（交错存储） yuv_image[height:height + height // 4, 0::2] = u_data.reshape(-1, 1)[0::2] yuv_image[height:height + height // 4, 1::2] = u_data.reshape(-1, 1)[1::2] yuv_image[height + height // 4:, 0::2] = v_data.reshape(-1, 1)[0::2] yuv_image[height + height // 4:, 1::2] = v_data.reshape(-1, 1)[1::2] # 使用 OpenCV 进行颜色转换 bgr_image = cv2.cvtColor(yuv_image, cv2.COLOR_YUV2BGR_I420) # 保存为 JPG 文件 cv2.imwrite(output_path, bgr_image) # 示例调用 yuv_to_jpg('rawvideo.yuv', 1920, 1080, 'output.jpg') ``` 上述代码实现了从指定路径读取 YUV 文件，并将其转换为 JPG 格式的功能。其中的关键在于正确处理 Y、U、V 分量的数据分布以及使用 OpenCV 提供的颜色空间转换函数。 --- ### 注意事项 - **输入参数说明** - `yuv_file_path`: 输入的 YUV 文件路径。 - `width`, `height`: 图像宽度和高度。 - `output_path`: 输出 JPG 文件的路径。 - **依赖库安装** 如果尚未安装所需的库，可以通过以下命令安装： ```bash pip install numpy opencv-python ``` - **兼容性** 上述代码假设输入的 YUV 文件采用 I420 编码模式（即平面 YUV）。如果使用的其他格式（如 NV12），则需调整代码逻辑以适配不同的采样结构[^2]。 ---