OpenCV CUDA模块中逐元素操作------数学函数

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

算法描述

在OpenCV的CUDA模块中,确实存在一系列用于执行逐元素数学运算的函数,包括指数、对数、平方根等。这些函数对于高级图像处理算法非常有用,比如计算图像梯度幅值时可能用到的平方根操作。

主要函数

1. 指数 - cv::cuda::exp

复制代码
描述:计算每个数组元素的自然指数(e的幂)。
#### 函数原型
cpp 复制代码
 void cv::cuda::exp
 (
        InputArray src, 
        OutputArray dst, 
        Stream& stream = Stream::Null()
    );
参数
  • src: 输入数组。
  • dst: 输出数组。
  • stream: CUDA流用于异步执行(可选)。

2. 对数 - cv::cuda::log

复制代码
描述:计算每个数组元素的自然对数。

函数原型:与cv::cuda::exp类似,但执行的是对数操作。

3. 平方根 - cv::cuda::sqrt

复制代码
描述:计算每个数组元素的平方根。
函数原型

void cv::cuda::sqrt

(

InputArray src,

OutputArray dst,

Stream& stream = Stream::Null()

);

参数
  • src: 输入数组。
  • dst: 输出数组。
  • stream: CUDA流用于异步执行(可选)。

代码示例

cpp 复制代码
#include <iostream>
#include <opencv2/cudaarithm.hpp>
#include <opencv2/opencv.hpp>

int main()
{
    // 创建一个测试图像(单通道浮点型)
    cv::Mat h_image = cv::Mat::ones( 512, 512, CV_32FC1 );
    h_image( cv::Rect( 100, 100, 100, 100 ) ).setTo( cv::Scalar( 0.5 ) );
    h_image( cv::Rect( 300, 300, 100, 100 ) ).setTo( cv::Scalar( 2.0 ) );

    // 上传到GPU
    cv::cuda::GpuMat d_image;
    d_image.upload( h_image );

    // 结果GPU矩阵
    cv::cuda::GpuMat d_exp_result, d_log_result, d_sqrt_result;

    // 指数运算
    cv::cuda::exp( d_image, d_exp_result );

    // 对数运算(加1防止log(0))
    cv::cuda::GpuMat d_temp;
    cv::cuda::add( d_image, cv::Scalar( 1.0 ), d_temp );  // 替代 d_image + 1.0
    cv::cuda::log( d_temp, d_log_result );

    // 平方根
    cv::cuda::sqrt( d_image, d_sqrt_result );

    // sin / cos 需要回到CPU执行
    cv::Mat h_sin_result, h_cos_result;

    // 先下载回CPU
    cv::Mat h_exp_result, h_log_result, h_sqrt_result;
    d_exp_result.download( h_exp_result );
    d_log_result.download( h_log_result );
    d_sqrt_result.download( h_sqrt_result );

    cv::Mat h_data;
    d_image.download( h_data );

    // 显示结果
    cv::imshow( "Original", h_data / 2.0 );
    cv::imshow( "Exp", h_exp_result );
    cv::imshow( "Log", h_log_result );
    cv::imshow( "Sqrt", h_sqrt_result );

    cv::waitKey( 0 );
    return 0;
}

运行结果

相关推荐
阿里云大数据AI技术6 分钟前
数据 + 模型 驱动 AI Native 应用发展
大数据·数据库·人工智能
大明哥_1 小时前
100 个 Coze 精品案例:Coze 全自动情感治愈视频混剪。用 Coze 工作流帮您节约 99% 的时间,从此告别手动!!
人工智能·agent
杨浦老苏1 小时前
IntelliSSH:AI驱动的远程服务器管理软件
人工智能·docker·ai·ssh·群晖
昵称是6硬币1 小时前
(DETR)End-to-End Object Detection with Transformers论文精读(逐段解析)
人工智能·深度学习·目标检测·计算机视觉·transformer
PyAIExplorer1 小时前
图像处理中的凸包检测:原理与实现
图像处理·人工智能
AndrewHZ1 小时前
【图像处理基石】如何检测到画面中的ppt并对其进行增强?
图像处理·人工智能·pytorch·opencv·目标检测·计算机视觉·图像增强
机器之心2 小时前
「Tokens是胡扯」,Mamba作者抛出颠覆性观点,揭露Transformer深层缺陷
人工智能
机器之心2 小时前
刚刚,马斯克发布Grok 4!全榜第一,年费飚到2万+
人工智能·grok
摆烂工程师2 小时前
教你在国内如何使用支付宝升级SuperGrok和Grok4的保姆绑卡教程
人工智能·支付宝·grok
fishjar1002 小时前
langgraph的ReAct应用
人工智能·ai