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

Posted 2022-11-20 振华OPPO

PC端	摄像头
Win10或者Ubuntu	400万像素USB模拟摄像头

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

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

C++版本代码

首先是定义视频文件的保存路径，然后创建并初始化摄像头对象。这里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();

Python版本代码

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()  # 释放窗口资源