博客

Python 读取 YUV(NV12) 视频文件实例

  • Post category:Python

YUV是一种常见的视频格式，其中NV12是一种常见的YUV格式。在Python中，可以使用OpenCV库来读取和处理YUV视频文件。下面是一个详细的攻略，包含两个示例，演示如何读取和处理YUV(NV12)视频文件。

示例一：使用OpenCV读取YUV(NV12)视频文件

下面是一个使用OpenCV库读取YUV(NV12)视频文件的示例：

import cv2

# 打开视频文件
cap = cv2.VideoCapture('test.yuv')

# 获取视频帧的宽度和高度
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))

# 循环读取视频帧
while True:
    # 读取一帧视频
    ret, frame = cap.read()

    # 如果视频读取完毕，则退出循环
    if not ret:
        break

    # 将YUV(NV12)格式的视频帧转换为BGR格式
    frame = cv2.cvtColor(frame, cv2.COLOR_YUV2BGR_NV12)

    # 显示视频帧
    cv2.imshow('frame', frame)

    # 等待用户按下q键退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# 释放视频文件和窗口资源
cap.release()
cv2.destroyAllWindows()

上面的代码演示了如何使用OpenCV库读取YUV(NV12)格式的视频文件。在读取视频帧后，需要将其转换为BGR格式，以便在窗口中显示。在循环中，cv2.imshow()函数显示视频帧，并使用cv2.waitKey()函数等待用户按下q键退出循环。最后，使用cap.release()函数释放视频文件资源，并使用cv2.destroyAllWindows()函数关闭窗口。

示例二：使用numpy库读取YUV(NV12)视频文件

下面是另一个使用numpy库读取YUV(NV12)视频文件的示例，它使用库来处理视频帧：

import cv2
import numpy as np

# 打开视频文件
cap = cv2.VideoCapture('test.yuv')

# 获取视频帧的宽度和高度
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))

# 循环读取视频帧
while True:
    # 读取一帧视频
    ret, frame = cap.read()

    # 如果视频读取完毕，则退出循环
    if not ret:
        break

    # 将YUV(NV12)格式的视频帧转换为BGR格式
    yuv = np.reshape(frame, (height * 3 // 2, width))
    bgr = cv2.cvtColor(yuv, cv2.COLOR_YUV2BGR_NV12)

    # 显示视频帧
    cv2.imshow('frame', bgr)

    # 等待用户按下q键退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# 释放视频文件和窗口资源
cap.release()
cv2.destroyAllWindows()

上面的代码与前面的示例类似，但使用了numpy库来处理视频帧。在读取视频后，使用np.reshape()函数将其转换为一个numpy数组，并使用cv2.cvtColor()函数将其转换为BGR格式。最后，使用cv2.imshow()函数显示视频帧，并使用cv2.waitKey()等待用户按下q键退出循环。

总结：本攻略演示了如何使用OpenCV和numpy库读取和处理YUV(NV12)视频文件。在读取视频帧后，需要将其转换为BGR格式，以便在窗口中显示。在循环中，cv2.imshow()函数显示视频帧，并使用cv2.waitKey()函数等待用户按下q键退出循环。最后，使用cap.release()函数释放视频文件资源，并使用cv2.destroyAllWindows()函数关闭窗口。

Tags: