python yuv转rgb

可以使用Python的Pillow库来进行YUV到RGB的转换。具体步骤如下：

1. 导入Pillow库

from PIL import Image

2. 加载YUV图像

with open('input.yuv', 'rb') as f:
    yuv_data = f.read()

# 假设图像大小为640x480，格式为YUV420P
y_size = 640 * 480
u_size = v_size = y_size // 4

y_data = yuv_data[:y_size]
u_data = yuv_data[y_size:y_size+u_size]
v_data = yuv_data[y_size+u_size:y_size+u_size+v_size]

# 将数据转换为PIL Image对象
y_image = Image.frombytes('L', (640, 480), y_data)
u_image = Image.frombytes('L', (320, 240), u_data)
v_image = Image.frombytes('L', (320, 240), v_data)

# 将U、V分量上采样到与Y分量相同的大小
u_image = u_image.resize((640, 480), resample=Image.BICUBIC)
v_image = v_image.resize((640, 480), resample=Image.BICUBIC)

# 将三个分量合并为一张图像
yuv_image = Image.merge('YCbCr', (y_image, u_image, v_image))

3. 将YUV图像转换为RGB图像

rgb_image = yuv_image.convert('RGB')

4. 保存RGB图像

rgb_image.save('output.jpg')

完整代码如下：

from PIL import Image

with open('input.yuv', 'rb') as f:
    yuv_data = f.read()

# 假设图像大小为640x480，格式为YUV420P
y_size = 640 * 480
u_size = v_size = y_size // 4

y_data = yuv_data[:y_size]
u_data = yuv_data[y_size:y_size+u_size]
v_data = yuv_data[y_size+u_size:y_size+u_size+v_size]

# 将数据转换为PIL Image对象
y_image = Image.frombytes('L', (640, 480), y_data)
u_image = Image.frombytes('L', (320, 240), u_data)
v_image = Image.frombytes('L', (320, 240), v_data)

# 将U、V分量上采样到与Y分量相同的大小
u_image = u_image.resize((640, 480), resample=Image.BICUBIC)
v_image = v_image.resize((640, 480), resample=Image.BICUBIC)

# 将三个分量合并为一张图像
yuv_image = Image.merge('YCbCr', (y_image, u_image, v_image))

# 将YUV图像转换为RGB图像
rgb_image = yuv_image.convert('RGB')

# 保存RGB图像
rgb_image.save('output.jpg')