### RGB到YUV420SP格式转换的Python实现 在计算机视觉和图像处理领域，颜色空间转换是一项常见的任务。OpenCV库中包含了许多用于颜色空间转换的功能，并且部分功能已经通过TBB和SSE进行了优化[^1]。然而，直接使用OpenCV进行RGB到YUV420SP（NV21）格式的转换需要手动实现，因为OpenCV并未直接提供此功能。 以下是一个完整的Python代码示例，展示如何将RGB图像转换为YUV420SP格式： ```python import numpy as np import cv2 def rgb_to_yuv420sp(rgb_image): # 确保输入图像为numpy数组并且是RGB格式 if rgb_image.ndim != 3 or rgb_image.shape[2] != 3: raise ValueError("Input image must be a 3-channel RGB image") # 将RGB图像转换为YUV颜色空间 yuv_image = cv2.cvtColor(rgb_image, cv2.COLOR_RGB2YUV) # 分离Y、U、V通道 y, u, v = cv2.split(yuv_image) # 调整U和V通道的分辨率以符合YUV420SP标准（缩小一半） u_resized = cv2.resize(u, (u.shape[1] // 2, u.shape[0] // 2)) v_resized = cv2.resize(v, (v.shape[1] // 2, v.shape[0] // 2)) # 将U和V通道交错存储（形成YUV420SP NV21格式） uv_interleaved = np.zeros((u_resized.shape[0], u_resized.shape[1] * 2), dtype=np.uint8) uv_interleaved[:, 0::2] = v_resized # V在偶数位置 uv_interleaved[:, 1::2] = u_resized # U在奇数位置 # 合并Y通道和UV交错通道 yuv420sp = np.vstack([y, uv_interleaved]) return yuv420sp # 示例：加载一张RGB图像并进行转换 rgb_image = cv2.imread('example.jpg') # 替换为实际的图像路径 if rgb_image is None: raise FileNotFoundError("Image file not found") # 转换为YUV420SP格式 yuv420sp_image = rgb_to_yuv420sp(rgb_image) # 保存或进一步处理结果 np.save('yuv420sp_image.npy', yuv420sp_image) # 保存为Numpy文件 ``` #### 代码说明 1. **颜色空间转换**：`cv2.cvtColor` 函数用于将RGB图像转换为YUV颜色空间。 2. **分辨率调整**：根据YUV420SP格式的要求，U和V通道的分辨率被缩小为原始尺寸的一半。 3. **通道交错**：YUV420SP格式要求U和V通道交错存储，其中V位于偶数位置，U位于奇数位置。 4. **合并输出**：最终的YUV420SP数据由Y通道和交错的UV通道组成。 #### 注意事项 - YUV420SP格式也被称为NV21格式，广泛应用于Android平台的相机API中[^2]。 - 如果需要处理其他格式（如NV12），只需调整U和V的交错顺序即可。