OpenCV保存USB摄像头视频流，附C++和Python源代码

刺骨的言语ヽ痛彻心扉 2023-10-02 18:11 26阅读 0赞

<table> 
 <thead> 
  <tr> 
   <th>PC端</th> 
   <th>摄像头</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td>Win10或者Ubuntu</td> 
   <td>400万像素USB模拟摄像头</td> 
  </tr> 
 </tbody> 
</table>

> USB摄像头是采用USB接口的安防摄像头，即插即用，傻瓜式操作，无需采集卡，无需电源，支持笔记本电脑。与传统的模拟摄像头相比成本更低，而且模数信号自动转换。方便实用，操作简单。关于相机的具体参数，还有数字/模拟信号，以及MPEG编解码，感兴趣的同学可以再查阅资料。

**首先笔记本一般都自带摄像头，所以插入USB摄像头后，识别设备编号为1，如果没有自带摄像头，capture()的参数为0。在Win10系统下录制视频分辨率使用640x480，在Ubuntu系统下分辨率采用相机的分辨率**

首先是定义视频文件的保存路径，然后创建并初始化摄像头对象。这里`VideoCapture capture(0);`也可以写成`VideoCapture capture; capture.open(0);`。然后创建写视频对象，设置保存视频信息，这里尤其注分辨率。然后就是输入流读入帧，再用输出流写到文件中。imshow显示每一帧图像，延迟10ms进入下一循环进行显示。最后退出程序时候将所有创建的对象和窗口进行销毁，防止内存泄漏。完整代码如下：

#include <iostream>
    #include <opencv2/opencv.hpp>
    using namespace std;
    using namespace cv;
    
    //视频文件保存位置
    string path = "/home/dzh/video.mp4";
    
    //打开摄像头
    VideoCapture capture(0);
    
    //写视频对象 
    VideoWriter outputVideo;
    
    //设置保存视频信息（路径，编码格式， 帧率， 分辨率）
    outputVideo.open(path, CV_FOURCC('M','P','4','V'), 30, Size(640, 480));
    
    // 如果视频没有打开成功
    if (!outputVideo.isOpened()) {
        
        cout << "fail to open!" << endl;
    }
    // 定义读取的帧
    Mat frame;
    // 定义想保存的视频帧数
    int count = 0;
    while(true) {
        
        //读取当前帧
        capture >> frame;
        count++;
        //显示当前视频帧
        imshow("output", frame);
        //写入当前帧
        outputVideo << frame;
        //任意键退出
        if ((waitKey(10) > 0) break;
        //或者录制600帧自动退出
        if (count >= 600) break;
    }
    // 释放写视频资源
    capture.release();
    outputVideo.release();
    // 销毁所有窗口
    destoryAllWindows();

<table> 
 <thead> 
  <tr> 
   <th><font>Python版本代码</font></th> 
  </tr> 
 </thead> 
</table>

Python代码的实现流程和C++相同，只是语法不同而已，这里扩展下编码格式的内容。

cv2.VideoWriter_fourcc('X', 'V', 'I', 'D') //该参数是MPEG-4编码类型，文件名后缀为.avi，可指定结果视频的大小

cv2.VideoWriter_fourcc('F', 'L', 'V', '1') //该参数是Flash视频，文件名后缀为.flv

cv2.VideoWriter_fourcc('X','2','6','4') //MPEG-4编码 .mp4 可指定结果视频的大小

完整的Python实现代码如下：

import cv2
    
    # 创建VideoWriter写视频流，编码格式16进制
    fourcc = cv2.VideoWriter_fourcc(*"mp4v")
    vw = cv2.VideoWriter("D:\\test.mp4", fourcc, 25, (640, 480))
    
    # 创建窗口
    cv2.namedWindow('video', cv2.WINDOW_NORMAL)
    cv2.resizeWindow('video', 1200, 720)  # 设置窗口大小
    
    # 获取视频设备
    cap = cv2.VideoCapture(0)
    # 从文件获取视频
    # cap = cv2.VideoCapture("文件地址")
    flag = 0
    # 判断摄像头是否已打开，若打开返回值则为True
    while cap.isOpened():
        # 从摄像头读视频帧
        ret, frame = cap.read()
        if ret == True:  # 判断是否读取到数据
            # 将视频帧在窗口中显示
            cv2.imshow('video', frame)
            # 写数据到多媒体文件
            key = cv2.waitKey(1)
            if flag == 1:
                #cv2.putText(frame, "image saving~", (100, 80), cv2.FONT_HERSHEY_SIMPLEX, 0.85, (0, 255, 0), 2, cv2.LINE_AA)
                cv2.imshow('video', frame)
                cv2.waitKey(1)
                vw.write(frame)
            if key == 32:  # 按空格键开始保存图像
                vw.write(frame)
                flag = 1
                print('开始保存图像')
            elif (key & 0xFF == ord('s')):  # 按 s 键停止保存图像
                flag = 0
                print('停止保存图像')
            elif (key & 0xFF == ord('q')):  # 按 q 键退出应用
                print('结束采集图像')
                break
        else:
            break
    # 释放资源
    cap.release()  # 释放VideoCapture
    vw.release()  # 释放VideoWrite
    cv2.destroyAllWindows()  # 释放窗口资源

![在这里插入图片描述][95490efbc6384d04bd72feea074769bb.png]

[95490efbc6384d04bd72feea074769bb.png]: https://img-blog.csdnimg.cn/95490efbc6384d04bd72feea074769bb.png