如何从相机中抓取YUV格式的视频,显示并处理它

import cv2

# 读取 YUV 视频
cap = cv2.VideoCapture(0)

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

    # 将 YUV 数据转换为 numpy 数组
    frame_yuv = cv2.cvtColor(frame, cv2.COLOR_BGR2YUV)

    # 显示 YUV 视频
    cv2.imshow('YUV', frame_yuv)

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

# 释放摄像头并关闭窗口
cap.release()
cv2.destroyAllWindows()

# 将 YUV 数据转换为 RGB 数据
frame_rgb = cv2.cvtColor(frame_yuv, cv2.COLOR_YUV2BGR)

# 将 YUV 数据转换为灰度数据
frame_gray = cv2.cvtColor(frame_yuv, cv2.COLOR_YUV2GRAY)

# 将 YUV 数据压缩
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
output_video = cv2.VideoWriter('output_yuv.mp4', fourcc, 30, (frame_width, frame_height))
output_video.write(frame_yuv)

# 将 YUV 数据转换为 JPEG 格式
fourcc = cv2.VideoWriter_fourcc(*'jpeg')
output_video = cv2.VideoWriter('output_yuv.jpg', fourcc, 30, (frame_width, frame_height))
output_video.write(frame_yuv)