OpenCV 的 CUDA 模块中用于将多个单通道的 GpuMat 图像合并成一个多通道的图像函数cv::cuda::merge

news2025/5/15 20:30:57

操作系统：ubuntu22.04
OpenCV版本：OpenCV4.9
IDE:Visual Studio Code
编程语言：C++11

算法描述

在 OpenCV 的 CUDA 模块中，cv::cuda::merge 函数用于将多个单通道的 GpuMat 图像合并成一个多通道的图像。该函数是 cv::merge 的 GPU 加速版本，适用于在 GPU 上高效执行图像通道合并操作。

函数原型

void cv::cuda::merge
(
    const GpuMat* src,        // 输入：指向多个单通道 GpuMat 的数组
    size_t n,                 // 输入：源数组中的通道数（通常为 3 或 4）
    OutputArray dst,          // 输出：合并后的多通道图像（GpuMat）
    Stream& stream = Stream::Null()  // 可选的 CUDA 流对象，默认使用默认流
)

参数说明

参数名	类型	说明
src	const GpuMat*	一个包含多个单通道 GpuMat 的数组，这些通道将被合并成一个多通道图像。所有输入矩阵必须具有相同的大小和类型。
n	size_t	输入数组中的通道数量。例如，合并 RGB 图像时为 3，RGBA 为 4。
dst	OutputArray	输出的合并结果图像，是一个多通道的 GpuMat。
stream	Stream&	可选的 CUDA 流对象，用于异步执行。如果不指定，则使用默认流（同步执行）。

使用示例

以下是一个从 CPU 将三个单通道图像上传到 GPU，并合并为一个三通道图像的完整示例：

#include <iostream>
#include <opencv2/cudaarithm.hpp>  // 包含 CUDA 算术运算头文件
#include <opencv2/opencv.hpp>

int main()
{
    // 读取一个三通道图像
    cv::Mat src_host = cv::imread( "/media/dingxin/data/study/OpenCV/sources/images/Lenna.png" );  // 替换为你自己的图片路径
    if ( src_host.empty() )
    {
        std::cerr << "无法读取图像" << std::endl;
        return -1;
    }

    // 将图像从BGR转换为三个单通道图像（蓝、绿、红）
    std::vector< cv::Mat > bgr_planes;
    cv::split( src_host, bgr_planes );

    // 创建用于存储GPU数据的GpuMat对象
    cv::cuda::GpuMat b_gpu, g_gpu, r_gpu;

    // 将每个通道上传到GPU
    b_gpu.upload( bgr_planes[ 0 ] );
    g_gpu.upload( bgr_planes[ 1 ] );
    r_gpu.upload( bgr_planes[ 2 ] );

    // 使用vector存储这些GpuMat对象
    std::vector< cv::cuda::GpuMat > src_vec = { b_gpu, g_gpu, r_gpu };

    // 目标多通道图像
    cv::cuda::GpuMat merged_gpu;

    // 合并成一个多通道图像
    cv::cuda::merge( src_vec, merged_gpu );

    // 下载结果以进行显示或保存
    cv::Mat merged_cpu;
    merged_gpu.download( merged_cpu );

    // 显示结果
    cv::imshow( "Merged Image", merged_cpu );
    cv::waitKey( 0 );

    return 0;
}