OpenCV CUDA模块图像处理-----对图像执行 均值漂移过程(Mean Shift Procedure)函数meanShiftProc()

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

算法描述

执行一个均值漂移过程(mean-shift procedure),并将处理后的点的信息(它们的颜色和位置)存储在两幅图像中。

该函数将源图像中的每个像素映射到一个新的颜色和位置,从而实现:

  • 颜色空间的聚类;
  • 空间位置的平滑;
  • 图像分割或风格化的预处理;

与 cv::cuda::meanShiftFiltering 不同的是,这个函数会输出两个结果:

  • 颜色空间的结果(dstr)
  • 空间位置的结果(dstsp)

函数原型

cpp 复制代码
void cv::cuda::meanShiftProc 	
(
 	InputArray  	src,
	OutputArray  	dstr,
	OutputArray  	dstsp,
	int  	sp,
	int  	sr,
	TermCriteria  	criteria = TermCriteria(TermCriteria::MAX_ITER+TermCriteria::EPS, 5, 1),
	Stream &  	stream = Stream::Null() 
) 		

参数

  • src 源图像。目前仅支持 CV_8UC4 类型的图像(即:8位无符号整型、4通道图像)。
  • dstr 目标图像,包含映射后点的颜色。其尺寸和类型与源图像相同。
  • dstsp 目标图像,包含映射后点的位置。其尺寸与源图像相同,类型为 CV_16SC2(即:16位有符号整型的二维坐标)。
  • sp 空间窗口半径(以像素为单位)。
  • sr 颜色窗口半径(以颜色差异为单位)。
  • criteria 终止条件。参见 TermCriteria 结构体,用于控制算法迭代的最大次数或收敛精度。
  • stream 用于异步版本的 CUDA 流(Stream)。

代码示例

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

int main()
{
    // Step 1: 读取图像并转换为 BGRA(4通道)
    cv::Mat h_src = cv::imread( "/media/dingxin/data/study/OpenCV/sources/images/Lenna.png" );
    if ( h_src.empty() )
    {
        std::cerr << "Failed to load image!" << std::endl;
        return -1;
    }

    cv::Mat h_rgba;
    cv::cvtColor( h_src, h_rgba, cv::COLOR_BGR2BGRA );

    // Step 2: 上传图像到 GPU
    cv::cuda::GpuMat d_src, d_str, d_sp;

    d_src.upload( h_rgba );

    // Step 3: 设置参数
    int sp = 10;  // 空间窗口大小
    int sr = 30;  // 颜色窗口大小

    // Step 4: 执行均值漂移过程
    cv::cuda::meanShiftProc( d_src, d_str, d_sp, sp, sr );

    // Step 5: 下载结果
    cv::Mat h_str, h_sp;
    d_str.download( h_str );  // 新颜色图像(CV_8UC4)
    d_sp.download( h_sp );    // 新空间坐标图像(CV_32FC4 或类似)

    // Step 6: 显示结果
    cv::imshow( "Original Image", h_src );
    cv::imshow( "Color Space Result", h_str );
    cv::waitKey( 0 );

    return 0;
}

运行结果

相关推荐
jndingxin34 分钟前
OpenCV中超分辨率(Super Resolution)模块类cv::dnn_superres::DnnSuperResImpl
人工智能·opencv·dnn
Trent19852 小时前
影楼精修-智能修图Agent
图像处理·人工智能·计算机视觉·aigc
叶子爱分享13 小时前
计算机视觉与图像处理的关系
图像处理·人工智能·计算机视觉
jndingxin17 小时前
OpenCV CUDA模块设备层-----反向二值化阈值处理函数thresh_binary_inv_func()
人工智能·opencv·计算机视觉
jndingxin18 小时前
OpenCV CUDA模块设备层-----在 GPU 上执行类似于 std::copy 的操作函数warpCopy()
人工智能·opencv·计算机视觉
晓131318 小时前
OpenCV篇——项目(二)OCR文档扫描
人工智能·python·opencv·pycharm·ocr
jndingxin18 小时前
OpenCV CUDA模块设备层-----在GPU 上高效地执行两个 uint 类型值的最大值比较函数vmax2()
人工智能·opencv·计算机视觉
加油吧zkf20 小时前
目标检测新纪元:DETR到Mamba实战解析
图像处理·人工智能·python·目标检测·分类
千宇宙航21 小时前
闲庭信步使用SV搭建图像测试平台:第二十七课——图像的腐蚀
图像处理·计算机视觉·fpga开发
luofeiju1 天前
RGB下的色彩变换:用线性代数解构色彩世界
图像处理·人工智能·opencv·线性代数