opencv cuda 保存视频数据C++案例

OpenCV 是一个开源的计算机视觉库，它提供了许多用于图像和视频处理的功能。CUDA（Compute Unified Device Architecture）是NVIDIA推出的一种并行计算平台和编程模型，它允许开发者利用GPU的强大计算能力来加速应用程序。

在C++中使用OpenCV和CUDA保存视频数据的基本步骤如下：

1. 包含必要的头文件：

#include <opencv2/opencv.hpp>
#include <opencv2/core/cuda.hpp>
#include <opencv2/cudaarithm.hpp>
#include <opencv2/cudacodec.hpp>

2. 初始化视频捕获设备：

cv::VideoCapture cap(0); // 打开默认摄像头
if (!cap.isOpened()) {
    std::cerr << "无法打开摄像头" << std::endl;
    return -1;
}

3. 设置视频编码器和输出文件：

int codec = cv::VideoWriter::fourcc('M', 'J', 'P', 'G');
double fps = 30.0;
cv::Size frame_size(640, 480);
cv::Ptr<cv::cudacodec::VideoWriter> d_writer = cv::cudacodec::createVideoWriter("output.avi", frame_size, codec, fps);

4. 循环捕获帧并写入文件：

cv::Mat frame;
cv::cuda::GpuMat d_frame;
while (true) {
    cap >> frame; // 从摄像头读取一帧
    if (frame.empty()) break;

    d_frame.upload(frame); // 上传到GPU
    d_writer->write(d_frame); // 写入视频文件

    // 显示帧
    cv::imshow("Frame", frame);
    if (cv::waitKey(1) == 27) break; // 按ESC退出
}

5. 释放资源：

cap.release();
cv::destroyAllWindows();

以上代码展示了如何使用OpenCV和CUDA在C++中捕获视频帧并将其保存到文件中。注意，这里使用了`cv::cudacodec::VideoWriter`来创建视频写入器，它是专门为使用CUDA优化的视频写入操作设计的。