OPenCV CUDA模块图像处理-----对图像执行 均值漂移滤波(Mean Shift Filtering)函数meanShiftFiltering()

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

算法描述

在 GPU 上对图像执行 均值漂移滤波(Mean Shift Filtering),用于图像分割或平滑处理。

该函数将输入图像中的每个像素用其邻域内颜色分布的"模式"代替,从而实现:

  • 颜色平滑(去除噪声)
  • 边界保持
  • 图像分割的预处理

函数原型

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

参数

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

代码示例

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

int main()
{
    // Step 1: 读取图像
    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;
    }

    // Step 2: 转换为 4 通道图像(BGRA)
    cv::cuda::GpuMat d_src, d_src_rgba;
    d_src.upload( h_src );
    cv::cuda::cvtColor( d_src, d_src_rgba, cv::COLOR_BGR2BGRA );  // 转为 CV_8UC4

    // Step 3: 创建输出图像
    cv::cuda::GpuMat d_dst;

    // Step 4: 设置参数并执行均值漂移滤波
    int sp = 10;  // 空间窗口大小
    int sr = 30;  // 颜色窗口大小
    cv::cuda::meanShiftFiltering( d_src_rgba, d_dst, sp, sr );

    // Step 5: 下载结果并显示
    cv::Mat h_dst;
    d_dst.download( h_dst );

    // Step 6: 如果需要恢复为 3 通道图像
    cv::cuda::GpuMat d_dst_bgr;
    cv::cuda::cvtColor( d_dst, d_dst_bgr, cv::COLOR_BGRA2BGR );

    cv::Mat h_final;
    d_dst_bgr.download( h_final );

    cv::imshow( "Original Image", h_src );
    cv::imshow( "Filtered Image", h_final );
    cv::waitKey( 0 );

    return 0;
}

运行结果

相关推荐
王尼莫啊4 小时前
【立体标定】圆形标定板标定python实现
开发语言·python·opencv
蓝桉8025 小时前
opencv学习(视频读取)
人工智能·opencv·学习
AndrewHZ6 小时前
【图像处理基石】如何对遥感图像进行实例分割?
图像处理·人工智能·python·大模型·实例分割·detectron2·遥感图像分割
dlraba8029 小时前
基于 OpenCV 与 sklearn 的数字识别:KNN 算法实践
opencv·算法·sklearn
德育处主任Pro10 小时前
『React』 组件通信全攻略
python·opencv·matplotlib
LeonDL16810 小时前
【通用视觉框架】基于QT+Halcon开发的流程拖拽式通用视觉框架软件,全套源码,开箱即用
图像处理·visionmaster·qt+halcon·流程拖拽式通用视觉框架软件·通用视觉框架软件·机器视觉通用框架·easyvision
一方热衷.16 小时前
YOLOv8/YOLOv11 C++ OpenCV DNN推理
c++·opencv·yolo
MUTA️1 天前
SAM附录详解
图像处理·人工智能·计算机视觉
PixelMind1 天前
【IQA技术专题】DISTS代码讲解
图像处理·人工智能·python·算法·iqa
FuckPatience1 天前
Winform C# 热力图制作要点
图像处理