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

Posted 振华OPPO

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了OpenCV保存USB摄像头视频流,附C++和Python源代码相关的知识,希望对你有一定的参考价值。

PC端摄像头
Win10或者Ubuntu400万像素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()  # 释放窗口资源

以上是关于OpenCV保存USB摄像头视频流,附C++和Python源代码的主要内容,如果未能解决你的问题,请参考以下文章

OpenCV保存摄像头视频和视频文件操作实战(附Python源码)

机器视觉行业实践技巧

机器视觉行业实践技巧 -- OpenCV技巧与方法:避坑指南

跪求用USB摄像头获取并保存图像的c或c++代码

OpenCV(C++版本)基础相关:图像的旋转视频文件摄像头使用视频处理与保存图像的直方图二维直方图

使用Python+Opencv从摄像头逐帧读取图片保存在本地