OpenCV CUDA模块设备层-----在 GPU 上执行类似于 std::copy 的操作函数warpCopy()

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

算法描述

OpenCV 的 CUDA 模块(cudev) 中的一个设备端内联模板函数,用于在 GPU 上执行类似于 std::copy 的操作,但专门针对 warp 规模的数据复制

该函数的作用是:

将一个范围内的元素从输入迭代器 beg 到 end 之间复制到输出迭代器 out 所指向的位置。

函数原型

cpp 复制代码
 template<class InIt , class OutIt >
__device__ __forceinline__ OutIt cv::cudev::warpCopy 	
( 	
	InIt  	beg,
	InIt  	end,
	OutIt  	out 
) 		

参数

  • InIt 输入迭代器类型(例如 PtrTraits<...>::ptr_type)
  • OutIt 输出迭代器类型(例如 PtrTraits<...>::ptr_type)

返回值

返回最终的输出迭代器 out,指向最后一个复制元素之后的位置,便于链式调用或后续操作。

使用场景

这个函数通常用于以下情况:

  • 在 CUDA kernel 中进行快速内存拷贝(如图像像素、数组等)
  • 实现自定义的图像变换或数据搬运逻辑
  • 构建更复杂的并行算法(如分块处理、扫描、归约)

它非常适合在每个线程负责多个数据项的场景下使用(即"warp-level"粒度的复制),可以提高内存访问效率和并行利用率。

代码

cpp 复制代码
#include <opencv2/opencv.hpp>
#include <opencv2/cudaimgproc.hpp>
#include <opencv2/cudev.hpp>

using namespace cv;
using namespace cv::cudev;

// 使用 warpCopy 的 kernel,用于高效复制一行像素
template <typename T>
__global__ void copyWarpCopyKernel(
    PtrStep<T> src,               // 注意:不是 const
    PtrStep<T> dst,
    int roiX, int roiY,
    int roiWidth, int roiHeight)
{
    int y = blockIdx.y * blockDim.y + threadIdx.y;

    if (y < roiHeight) {
        T* srcRow = &src(roiY + y, roiX);  // 正确获取非 const 指针
        T* dstRow = &dst(y, 0);

        warpCopy(srcRow, srcRow + roiWidth, dstRow);
    }
}

int main() {
    // 加载图像(灰度图)
    Mat h_src = imread("/media/dingxin/data/study/OpenCV/sources/images/Lenna.png", IMREAD_GRAYSCALE);
    if (h_src.empty()) {
        std::cerr << "Failed to load image!" << std::endl;
        return -1;
    }

    // 设置 ROI 参数
    int roiX = 100;
    int roiY = 50;
    int roiWidth = 320;
    int roiHeight = 240;

    // 上传到 GPU
    cuda::GpuMat d_src, d_dst;
    d_src.upload(h_src);
    d_dst.create(roiHeight, roiWidth, d_src.type());

    // 配置 kernel 参数(仅在 Y 方向并行)
    dim3 block(16, 16);
    dim3 grid(1, (roiHeight + block.y - 1) / block.y);

    // 启动 kernel
    copyWarpCopyKernel<uchar><<<grid, block>>>(d_src, d_dst, roiX, roiY, roiWidth, roiHeight);

    // 下载结果
    Mat h_dst;
    d_dst.download(h_dst);

    // 显示结果
    imshow("Copied ROI", h_dst);
    waitKey(0);

    return 0;
}

运行结果

相关推荐
Allen_LVyingbo18 分钟前
数智读书笔记系列035《未来医疗:医疗4.0引领第四次医疗产业变革》
人工智能·经验分享·笔记·健康医疗
zzc92123 分钟前
时频图数据集更正程序,去除坐标轴白边及调整对应的标签值
人工智能·深度学习·数据集·标签·时频图·更正·白边
isNotNullX25 分钟前
什么是数据分析?常见方法全解析
大数据·数据库·数据仓库·人工智能·数据分析
riveting34 分钟前
明远智睿H618:开启多场景智慧生活新时代
人工智能·嵌入式硬件·智能硬件·lga封装·3506
夜阑卧听风吹雨,铁马冰河入梦来1 小时前
Spring AI 阿里巴巴学习
人工智能·学习·spring
c7691 小时前
【文献笔记】Automatic Chain of Thought Prompting in Large Language Models
人工智能·笔记·语言模型·论文笔记
Blossom.1181 小时前
机器学习在智能供应链中的应用:需求预测与物流优化
人工智能·深度学习·神经网络·机器学习·计算机视觉·机器人·语音识别
Gyoku Mint2 小时前
深度学习×第4卷:Pytorch实战——她第一次用张量去拟合你的轨迹
人工智能·pytorch·python·深度学习·神经网络·算法·聚类
zzywxc7872 小时前
AI大模型的技术演进、流程重构、行业影响三个维度的系统性分析
人工智能·重构
点控云2 小时前
智能私域运营中枢:从客户视角看 SCRM 的体验革新与价值重构
大数据·人工智能·科技·重构·外呼系统·呼叫中心