九、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/bmpif (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);// 创建 VideoWriterVideoWriter 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;}// 尺寸不一致就 resizeif (img.size() != frameSize) {resize(img, img, frameSize);}writer.write(img);}writer.release();cout <<"Video saved as " << outputVideo << endl;return 0;}