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 小时前
【学习】Linux 内核中的 cgroup freezer 子系统
人工智能·opencv·计算机视觉
这张生成的图像能检测吗7 小时前
(论文速读)RandAR:突破传统限制的随机顺序图像自回归生成模型
图像处理·人工智能·机器学习·计算机视觉·生成模型·自回归模型
星期天要睡觉1 天前
计算机视觉(opencv)实战六——图像形态学(腐蚀、膨胀、开运算、闭运算、梯度、顶帽、黑帽)
人工智能·opencv·计算机视觉
hixiong1231 天前
用OpencvSharp编写视频录制工具
opencv·c#·音视频
房开民1 天前
基于OpenCV的物体识别与计数
人工智能·opencv·计算机视觉
Ronin-Lotus1 天前
图像处理篇---掩膜获取
图像处理·人工智能
小关会打代码1 天前
计算机视觉第一课opencv(三)保姆级教学
人工智能·opencv·计算机视觉·边缘检测·膨胀·腐蚀
山烛1 天前
OpenCV图像形态学操作
图像处理·人工智能·python·opencv·计算机视觉·图像形态学
hixiong1231 天前
C# OpencvSharp获取Astra Pro奥比中光深度相机深度图
数码相机·opencv·计算机视觉·c#
qq_526099131 天前
六口网卡:工业级千兆以太网控制器,赋能智能制造与边缘计算
图像处理·自动化