如何 OPENCV+CUDA+VideoCapture？答案

【问题标题】：Howto OPENCV+CUDA+VideoCapture?如何 OPENCV+CUDA+VideoCapture？
【发布时间】：2020-09-20 17:37:57
【问题描述】：

我有一台带有 2 个 Nvidia RTX GPU 的机器。主要任务是解码来自 IP 摄像机的 h264 视频并将原始帧编码为 JPEG使用 GPU。

我从以下来源构建了 opencv + cuda：

cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local -D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib/modules -D WITH_CUDA=ON -D WITH_TBB=ON -D ENABLE_FAST_MATH=1 -D CUDA_FAST_MATH=1 -D WITH_CUBLAS=1 -D WITH_QT=OFF -D WITH_NVCUVID -D WITH_NVCUVID=ON -D CMAKE_C_COMPILER=/usr/bin/gcc-6 ..
make
sudo make install

我有一个简单的程序可以捕获 RTSP 并保存 JPEG：

#include <strings.h>
#include <stdio.h>
#include <opencv2/opencv.hpp>
#include <opencv2/core/cuda.hpp>

using namespace std;

int main(int argc, char** argv)
{
    if ( argc != 2 ) {
      printf("no args\n");
      return -1;
    }
    cv::Mat frame;
    cv::VideoCapture cap;
    cap.open(argv[1]);
    if (!cap.isOpened()) {
        printf("ERROR! Unable to open camera\n");
        return -1;
    }
    printf("Start grabbing\n");
    for (;;)
    {
        cap.read(frame);
        if (frame.empty()) {
            printf("ERROR! blank frame grabbed\n");
            break;
        }
        vector<int> p;
        vector<unsigned char> buf;
        p.push_back(cv::IMWRITE_JPEG_QUALITY);
        p.push_back(50);
        imencode(".jpg", frame, buf, p);
        // Do something with buf ......
        printf("READY JPEG\n");
    }
    return 0;
}

CMakeLists.txt：

cmake_minimum_required(VERSION 2.8)
project( test_jpeg_cpu )
find_package( OpenCV REQUIRED )
find_package( CUDA REQUIRED )
include_directories( ${OpenCV_INCLUDE_DIRS} ${CUDA_INCLUDE_DIRS} )
add_executable( test_jpeg_cpu test_jpeg_cpu.cpp )
target_link_libraries( test_jpeg_cpu ${OpenCV_LIBS} ${CUDA_LIBRARIES} cuda ${CMAKE_DL_LIBS} )

但它只使用 CPU。为了让这个程序使用 GPU，我需要做些什么改变？

提前致谢。

【问题讨论】：

也许从这里开始阅读：opencv.org/platforms/cuda 您需要使用其他 Mat 类型和其他标题

标签： c++ opencv video gpu

【解决方案1】：

您正在使用imwrite API，它在后端使用libjpeg 进行基于CPU 的编码。如果您想使用 GPU，则需要使用 cv::cudacodec::createVideoWriter API（假设它已内置到您的 OpenCV 安装中）。

OpenCV 源代码库中有example code for GPU-accelerated video encoding。

【讨论】：