OpenCV CUDA 模块中用于在 GPU 上计算矩阵中每个元素的绝对值或复数的模函数abs()

jndingxin2025-05-17 15:49

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

算法描述

void cv::cuda::abs(InputArray src, OutputArray dst, Stream &stream = Stream::Null()) 是 OpenCV 的 CUDA 模块中的一个函数，用于在 GPU 上计算矩阵中每个元素的绝对值或复数的模。这个函数可以加速图像处理和计算机视觉任务中对绝对值的操作。

函数原型

cpp 复制代码

void cv::cuda::abs
(
    InputArray src,           // 输入 GPU 矩阵
    OutputArray dst,          // 输出 GPU 矩阵
    Stream& stream = Stream::Null()  // 可选的 CUDA 流（默认为同步执行）
)

参数

参数名	类型	描述
src	InputArray	输入的 GPU 矩阵（单通道），支持类型包括：CV_32FC1、CV_32FC2、CV_64FC2。
}dst	OutputArray	输出结果矩阵，大小与输入相同，数据类型为 CV_32F。
stream	Stream&	可选参数，指定 CUDA 流以实现异步执行，默认是同步执行。

代码示例

cpp 复制代码

#include <opencv2/cudaarithm.hpp>  // 包含 CUDA 图像处理函数
#include <opencv2/opencv.hpp>

int main()
{
    // 创建 CPU 矩阵并初始化
    cv::Mat h_src = ( cv::Mat_< float >( 2, 2 ) << -1.0f, 2.0f, -3.0f, 4.0f );

    // 上传到 GPU
    cv::cuda::GpuMat d_src;
    d_src.upload( h_src );

    // 创建输出 GPU 矩阵
    cv::cuda::GpuMat d_dst;

    // 调用 CUDA 版本的 abs 函数
    cv::cuda::abs( d_src, d_dst );

    // 将结果下载回 CPU
    cv::Mat h_dst;
    d_dst.download( h_dst );

    // 打印结果
    std::cout << "绝对值结果：\n" << h_dst << std::endl;

    return 0;
}

运行结果

bash 复制代码

绝对值结果：
[1, 2;
 3, 4]