九、OpenCV中视频的录制

文章目录

• • [一、OpenCV 视频录制概述](#一、OpenCV 视频录制概述)
  • 二、示例代码

一、OpenCV 视频录制概述

在 OpenCV 中，视频录制主要靠 cv::VideoWriter 类。它可以把连续的 cv::Mat 帧写入一个视频文件（如 .avi、.mp4 等）。

VideoWriter类

构造函数：

VideoWriter::VideoWriter(const String &filename,
                         int fourcc,
                         double fps,
                         Size frameSize,
                         bool isColor = true);

参数说明：

• filename：保存的视频文件名，例如 "output.avi"
• fourcc：视频编码方式，如 CV_FOURCC('M','J','P','G') 或 VideoWriter::fourcc('M','J','P','G')
• fps：帧率（帧每秒）
• frameSize：视频每帧的尺寸，Size(width, height)
• isColor：是否为彩色视频，true 彩色，false 灰度

常用编码（fourcc）：

int fourcc = VideoWriter::fourcc('M','J','P','G'); // AVI 常用
int fourcc_mp4 = VideoWriter::fourcc('a','v','c','1'); // MP4 H264

二、示例代码

示例 1：从摄像头录制视频

#include <opencv2/opencv.hpp>
#include <iostream>
using namespace cv;
using namespace std;

int main() {
    VideoCapture cap(0); // 打开摄像头
    if (!cap.isOpened()) {
        cout << "Cannot open camera" << endl;
        return -1;
    }

    // 视频参数
    int frame_width = static_cast<int>(cap.get(CAP_PROP_FRAME_WIDTH));
    int frame_height = static_cast<int>(cap.get(CAP_PROP_FRAME_HEIGHT));
    VideoWriter writer("output.avi",
                       VideoWriter::fourcc('M','J','P','G'),
                       30, Size(frame_width, frame_height));

    Mat frame;
    while (true) {
        cap >> frame;
        if (frame.empty()) break;

        writer.write(frame); // 写入视频

        imshow("Camera", frame);
        if (waitKey(1) == 27) break; // 按 ESC 停止
    }

    cap.release();
    writer.release();
    destroyAllWindows();
    return 0;
}

示例 2：将视频文件处理后保存

#include <opencv2/opencv.hpp>
#include <iostream>
using namespace cv;
using namespace std;

int main() {
    VideoCapture cap("opencv_demo.mp4");
    if (!cap.isOpened()) return -1;

    int width = cap.get(CAP_PROP_FRAME_WIDTH);
    int height = cap.get(CAP_PROP_FRAME_HEIGHT);
    VideoWriter writer("output_processed.avi",
        VideoWriter::fourcc('M', 'J', 'P', 'G'),
        30, Size(width, height));

    Mat frame;
    while (true) {
        cap >> frame;
        if (frame.empty()) break;

        // 处理帧，例如灰度
        cvtColor(frame, frame, COLOR_BGR2GRAY);
        cvtColor(frame, frame, COLOR_GRAY2BGR); // 保持三通道以写入

        writer.write(frame);
    }
    cap.release();
    writer.release();
    return 0;
}

示例3：自动读取指定文件夹下所有图片生成视频

#include <opencv2/opencv.hpp>
#include <iostream>
#include <filesystem>
#include <vector>
#include <algorithm>
using namespace cv;
using namespace std;
namespace fs = std::filesystem;

int main() {
    string folderPath = "images"; // 图片文件夹路径
    string outputVideo = "output_video.avi";
    double fps = 30.0; // 帧率

    vector<string> imgFiles;

    // 遍历文件夹，收集图片路径
    for (const auto& entry : fs::directory_iterator(folderPath)) {
        if (entry.is_regular_file()) {
            string ext = entry.path().extension().string();
            // 支持 jpg/png/bmp
            if (ext == ".jpg" || ext == ".png" || ext == ".bmp")
                imgFiles.push_back(entry.path().string());
        }
    }

    if (imgFiles.empty()) {
        cout << "No images found in folder!" << endl;
        return -1;
    }

    // 按文件名排序
    sort(imgFiles.begin(), imgFiles.end());

    // 读取第一张图片获取尺寸
    Mat firstImg = imread(imgFiles[0]);
    if (firstImg.empty()) {
        cout << "Failed to read " << imgFiles[0] << endl;
        return -1;
    }
    Size frameSize(firstImg.cols, firstImg.rows);

    // 创建 VideoWriter
    VideoWriter writer(outputVideo,
                       VideoWriter::fourcc('M','J','P','G'),
                       fps,
                       frameSize,
                       true);

    if (!writer.isOpened()) {
        cout << "Cannot open VideoWriter!" << endl;
        return -1;
    }

    // 写入每张图片
    for (const auto& file : imgFiles) {
        Mat img = imread(file);
        if (img.empty()) {
            cout << "Failed to read " << file << endl;
            continue;
        }

        // 尺寸不一致就 resize
        if (img.size() != frameSize) {
            resize(img, img, frameSize);
        }

        writer.write(img);
    }

    writer.release();
    cout << "Video saved as " << outputVideo << endl;
    return 0;
}