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;
}

运行结果

相关推荐
李元豪32 分钟前
【知足常乐ai笔记】机器人强化学习
人工智能·笔记·机器人
沫儿笙33 分钟前
焊接机器人智能节气装置
人工智能·机器人
MidJourney中文版34 分钟前
老年人与机器人玩具的情感连接
人工智能·机器人·语音识别
Codebee35 分钟前
AI驱动的低代码革命:解构与重塑开发范式
人工智能·低代码·代码规范
数据库安全37 分钟前
首批|美创智能数据安全分类分级平台获CCIA“网络安全新产品”
大数据·人工智能·web安全
Dymc38 分钟前
【目标检测之Ultralytics预测框颜色修改】
人工智能·yolo·目标检测·计算机视觉
HuashuiMu花水木1 小时前
PyTorch笔记3----------统计学相关函数
人工智能·pytorch·笔记
m0_743106461 小时前
【论文笔记】BlockGaussian:巧妙解决大规模场景重建中的伪影问题
论文阅读·计算机视觉·3d·aigc·几何学
算家计算1 小时前
6 亿参数玩转 20 + 语言!OuteTTS-1.0-0.6B本地部署教程,轻量模型也能 hold 住跨语言合成
人工智能·开源
柠檬味拥抱1 小时前
USB‑C 式的工具联接:MCP 的模块化及通用标准探讨
人工智能