Skip to main content

USB摄像头视频拉流

摄像头拉流

USB摄像头视频拉流​

  1. 将USB摄像头插入Atlas 200I DK A2 开发者套件的USB接口。

  2. 使用摄像头逐帧采集，并将推理结果写入结果视频中。

    示例代码如下。

    import cv2

    cap = cv2.VideoCapture(0)

    # 获取保存视频相关变量
    fps = cap.get(cv2.CAP_PROP_FPS)  # 获取帧数
    fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # 设置视频编码类型
    outfile = 'video_result.mp4'  # 设置结果保存文件
    video_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))  # 获取原视频的高度
    video_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))  # 获取原视频的宽度
    writer = cv2.VideoWriter(outfile, fourcc, fps, (video_width, video_height))  # 获取视频写入writer

    try:
        while(cap.isOpened()):  # 在摄像头打开的情况下循环执行
            ret, frame = cap.read()  # 此处 frame 为 bgr 格式图片

            # ========================================
            # 这里放入模型前处理、推理、后处理相关代码，根据实际情况修改
            img = preprocess(frame)  # 前处理
            infer_res = infer(img)  # 推理
            img_res = postprocess(infer_res)  # 后处理
            # ========================================

            # 将推理结果写入视频
            writer.write(img_res)  
    except KeyboardInterrupt:
        cap.release()  # 释放摄像头资源
        writer.release()  # 释放视频写入资源
    finally:
        cap.release()
        writer.release()

    :::note 说明 此代码为样例代码，用户需自行修改模型的前处理、推理、后处理部分。 :::

Atlas 200I DK A2开发者套件

摄像头拉流

在线提单