samples/cpp/videowriter_basic.cpp

使用 VideoCapture 和 VideoWriter 类的示例

 
#include <opencv2/core.hpp>
#include <opencv2/videoio.hpp>
#include <opencv2/highgui.hpp>
#include <iostream>
#include <stdio.h>
 
using namespace cv;
using namespace std;
 
int main(int, char**)
{
    Mat src;
    // 使用默认摄像头作为视频源
    VideoCapture cap(0);
    // 检查是否成功
    if (!cap.isOpened()) {
cerr << "错误！无法打开摄像头\n";
        return -1;
    }
    // 从摄像头获取一帧以了解帧大小和类型
cap >> src;
    // 检查是否成功
    if (src.empty()) {
cerr << "错误！抓取的帧为空\n";
        return -1;
    }
    bool isColor = (src.type() == CV_8UC3);
 
    //--- 初始化 VIDEOWRITER
    VideoWriter writer;
    int codec = VideoWriter::fourcc('M', 'J', 'P', 'G'); // 选择所需的编解码器 (运行时必须可用)
    double fps = 25.0; // 创建的视频流的帧率
    string filename = "./live.avi"; // 输出视频文件的名称
writer.open(filename, codec, fps, src.size(), isColor);
    // 检查是否成功
    if (!writer.isOpened()) {
cerr << "无法打开输出视频文件进行写入\n";
        return -1;
    }
 
    //--- 抓取和写入循环
cout << "正在写入视频文件: " << filename << endl
<< "按任意键终止" << endl;
    for (;;)
    {
        // 检查是否成功
        if (!cap.read(src)) {
cerr << "错误！抓取的帧为空\n";
            break;
        }
        // 将帧编码到视频文件流中
writer.write(src);
        // 显示实时画面并等待按键，超时时间足够长以显示图像
imshow("Live", src);
        if (waitKey(5) >= 0)
            break;
    }
    // 视频文件将在 VideoWriter 析构函数中自动关闭和释放
    return 0;
}