python读取yuv文件示例代码

准备工作

首先，确保你已经安装了Python和OpenCV库。如果尚未安装，可以通过以下命令安装：

pip install opencv-python

YUV文件格式简介

YUV是一种将亮度（Y）和色度（U、V）分离的颜色编码格式。在YUV420格式中，每个像素的亮度值（Y）单独存储，而色度值（U、V）则按照一定规则进行抽样和存储。

代码实现

我们使用Python编写一个简单的脚本来读取和显示YUV文件。以下是示例代码：

# 导入必要的库
import cv2
import numpy as np

class VideoCaptureYUV:
    def __init__(self, filename, size):
        self.height, self.width = size
        self.frame_len = self.width * self.height * 3 / 2
        self.f = open(filename, 'rb')
        self.shape = (int(self.height*1.5), self.width)

    def read_raw(self):
        try:
            raw = self.f.read(self.frame_len)
            yuv = np.frombuffer(raw, dtype=np.uint8)
            yuv = yuv.reshape(self.shape)
        except Exception as e:
            print(str(e))
            return False, None
        return True, yuv

    def read(self):
        ret, yuv = self.read_raw()
        if not ret:
            return ret, yuv
        bgr = cv2.cvtColor(yuv, cv2.COLOR_YUV2BGR_I420)
        return ret, bgr

if __name__ == "__main__":
    # 设置YUV文件路径和大小
    filename = "akiyo_cif.yuv"
    size = (288, 352)

    # 创建YUV视频捕获对象
    cap = VideoCaptureYUV(filename, size)

    # 循环读取并显示视频帧
    while True:
        ret, frame = cap.read()
        if ret:
            cv2.imshow("frame", frame)
            cv2.waitKey(30)  # 设置帧间延迟
        else:
            break

    # 释放资源
    cap.f.close()
    cv2.destroyAllWindows()

解释代码

• VideoCaptureYUV类：这是一个自定义的视频捕获类，用于读取YUV文件。它包括了初始化方法和读取方法，能够从YUV文件中读取原始数据并转换为BGR格式用于显示。
• 主程序：在主程序中，我们指定了要处理的YUV文件名和视频帧的大小，并创建了一个VideoCaptureYUV对象。然后，使用一个无限循环读取每一帧，将其转换为BGR格式，并使用OpenCV的imshow函数显示在屏幕上，通过waitKey设置了帧之间的延迟时间，最后通过destroyAllWindows关闭窗口和释放资源。