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

OpenCV CUDA模块中逐元素操作------算术运算

news 来源：原创 2025/5/15 6:34:00

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

算法描述

算术运算：包括加法、减法、乘法和除法等基本运算。这些操作可以应用于两个相同尺寸的矩阵，或者一个矩阵与一个标量之间。例如，将两个图像矩阵相加可以用于图像合成。

主要函数

1. 加法操作 (cv::cuda::add)

功能：对两个矩阵或一个矩阵和一个标量执行逐元素加法操作。

函数原型

void cv::cuda::add
(InputArray src1,        // 第一个输入矩阵或标量InputArray src2,        // 第二个输入矩阵或标量OutputArray dst,       // 输出矩阵const GpuMat& mask = GpuMat(), // 可选掩码矩阵int dtype = -1,        // 输出矩阵的数据类型（默认与输入相同）Stream& stream = Stream::Null() // 可选CUDA流
);

参数介绍

src1: 输入的第一个矩阵或标量。
src2: 输入的第二个矩阵或标量。
dst: 输出的结果矩阵。
mask: 可选的掩码矩阵，用于指定哪些像素参与计算，默认为空（不使用掩码）。
dtype: 输出矩阵的数据类型，如果设置为-1，则与输入矩阵相同。
stream: CUDA流对象，用于控制操作的异步执行，默认为同步执行。

2. 减法操作 (cv::cuda::subtract)

功能：对两个矩阵或一个矩阵和一个标量执行逐元素减法操作。

函数原型

void cv::cuda::subtract
(InputArray src1,InputArray src2,OutputArray dst,const GpuMat& mask = GpuMat(),int dtype = -1,Stream& stream = Stream::Null()
);

参数介绍

src1: 输入的第一个矩阵或标量。
src2: 输入的第二个矩阵或标量。
dst: 输出的结果矩阵。
mask: 可选的掩码矩阵，用于指定哪些像素参与计算，默认为空（不使用掩码）。
dtype: 输出矩阵的数据类型，如果设置为-1，则与输入矩阵相同。
stream: CUDA流对象，用于控制操作的异步执行，默认为同步执行。

3. 乘法操作 (cv::cuda::multiply)

功能：对两个矩阵或一个矩阵和一个标量执行逐元素乘法操作。

函数原型

void cv::cuda::multiply
(InputArray src1,InputArray src2,OutputArray dst,double scale = 1,      // 缩放因子，默认为1int dtype = -1,        // 输出矩阵的数据类型（默认与输入相同）Stream& stream = Stream::Null()
);

参数

src1: 输入的第一个矩阵或标量。
src2: 输入的第二个矩阵或标量。
dst: 输出的结果矩阵。
scale: 缩放因子，结果将乘以这个值，默认为1。
dtype: 输出矩阵的数据类型，如果设置为-1，则与输入矩阵相同。
stream: CUDA流对象，用于控制操作的异步执行，默认为同步执行。

4. 除法操作 (cv::cuda::divide)

功能：对两个矩阵或一个矩阵和一个标量执行逐元素除法操作。

函数原型

void cv::cuda::divide
(InputArray src1,InputArray src2,OutputArray dst,double scale = 1,      // 缩放因子，默认为1int dtype = -1,        // 输出矩阵的数据类型（默认与输入相同）Stream& stream = Stream::Null()
);

参数

src1: 输入的第一个矩阵或标量。
src2: 输入的第二个矩阵或标量。
dst: 输出的结果矩阵。
scale: 缩放因子，结果将乘以这个值，默认为1。
dtype: 输出矩阵的数据类型，如果设置为-1，则与输入矩阵相同。
stream: CUDA流对象，用于控制操作的异步执行，默认为同步执行。

5. 绝对值 (cv::cuda::abs)

功能：计算每个元素的绝对值，支持实数和复数类型。

函数原型

void cv::cuda::abs
(InputArray src,        // 输入矩阵OutputArray dst,       // 输出矩阵Stream& stream = Stream::Null()
);

参数

src: 输入的矩阵。
dst: 输出的结果矩阵。
stream: CUDA流对象，用于控制操作的异步执行，默认为同步执行。

6. 绝对差值 (cv::cuda::absdiff)

功能：计算两个数组对应元素差值的绝对值。

函数原型

void cv::cuda::absdiff
(InputArray src1,InputArray src2,OutputArray dst,Stream& stream = Stream::Null()
);

参数

src1: 输入的第一个矩阵。
src2: 输入的第二个矩阵。
dst: 输出的结果矩阵。
stream: CUDA流对象，用于控制操作的异步执行，默认为同步执行。

代码示例

#include <opencv2/cudaarithm.hpp>  // 包含CUDA算术运算头文件
#include <opencv2/opencv.hpp>int main()
{// 创建两个输入矩阵cv::Mat h_src1 = ( cv::Mat_< float >( 3, 3 ) << 1, 2, 3, 4, 5, 6, 7, 8, 9 );  // 第一个输入矩阵cv::Mat h_src2 = ( cv::Mat_< float >( 3, 3 ) << 9, 8, 7, 6, 5, 4, 3, 2, 1 );  // 第二个输入矩阵// 将它们上传到GPUcv::cuda::GpuMat d_src1, d_src2;d_src1.upload( h_src1 );d_src2.upload( h_src2 );// 定义输出矩阵cv::cuda::GpuMat d_dst;// 加法操作cv::cuda::add( d_src1, d_src2, d_dst );cv::Mat h_add;d_dst.download( h_add );std::cout << "Result (Addition):\n" << h_add << std::endl;// 减法操作cv::cuda::subtract( d_src1, d_src2, d_dst );cv::Mat h_subtract;d_dst.download( h_subtract );std::cout << "Result (Subtraction):\n" << h_subtract << std::endl;// 乘法操作cv::cuda::multiply( d_src1, d_src2, d_dst, 1.0, CV_32F );cv::Mat h_multiply;d_dst.download( h_multiply );std::cout << "Result (Multiplication):\n" << h_multiply << std::endl;// 除法操作cv::cuda::divide( d_src1, d_src2, d_dst, 1.0, CV_32F );cv::Mat h_divide;d_dst.download( h_divide );std::cout << "Result (Division):\n" << h_divide << std::endl;// 绝对差值cv::cuda::absdiff( d_src1, d_src2, d_dst );cv::Mat h_absdiff;d_dst.download( h_absdiff );std::cout << "Result (AbsDiff):\n" << h_absdiff << std::endl;return 0;
}

运行结果

Result (Addition):
[10, 10, 10;10, 10, 10;10, 10, 10]
Result (Subtraction):
[-8, -6, -4;-2, 0, 2;4, 6, 8]
Result (Multiplication):
[9, 16, 21;24, 25, 24;21, 16, 9]
Result (Division):
[0.11111111, 0.25, 0.42857143;0.66666669, 1, 1.5;2.3333333, 4, 9]
Result (AbsDiff):
[8, 6, 4;2, 0, 2;4, 6, 8]