当前位置：首页 > news >正文

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

news 来源：原创 2025/5/15 5:42:28

操作系统：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 上计算矩阵中每个元素的绝对值或复数的模。这个函数可以加速图像处理和计算机视觉任务中对绝对值的操作。

函数原型

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 流以实现异步执行，默认是同步执行。

代码示例

#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 );// 上传到 GPUcv::cuda::GpuMat d_src;d_src.upload( h_src );// 创建输出 GPU 矩阵cv::cuda::GpuMat d_dst;// 调用 CUDA 版本的 abs 函数cv::cuda::abs( d_src, d_dst );// 将结果下载回 CPUcv::Mat h_dst;d_dst.download( h_dst );// 打印结果std::cout << "绝对值结果：\n" << h_dst << std::endl;return 0;
}