深入解析：OpenCV CUDA模块霍夫变换------在 GPU 上执行概率霍夫变换检测图像中的线段端点类cv::cuda::HoughSegmentDetector

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

算法描述

cv::cuda::HoughSegmentDetector 是 OpenCV 的 CUDA 模块中一个非常重要的类，它用于在 GPU 上执行 概率霍夫变换（Probabilistic Hough Transform），能够检测图像中的线段端点（即直线段），而不是标准霍夫变换中表示为 (rho, theta) 的无限长直线。

类定义

class cv::cuda::HoughSegmentDetector :
public cv::cuda::Algorithm

继承自 cv::cuda::Algorithm，提供了 GPU 加速的线段检测功能。

主要功能

• 在 GPU 上进行 概率霍夫变换
• 输出为一系列 线段端点，格式为 Vec4i(x1, y1, x2, y2)
• 支持边缘图作为输入（通常是 Canny 边缘检测后的图像）

构造函数 & 创建方式

你可以通过以下方式创建该类的对象：

cv::Ptr<cv::cuda::HoughSegmentDetector> hough =
  cv::cuda::createHoughSegmentDetector(rho, theta, threshold, minLineLength, maxLineGap)
  ;

参数说明：

参数名	类型	含义
rho	float	距离分辨率（像素）
theta	float	角度分辨率（弧度）
threshold	int	投票阈值，只有投票数大于等于此值的线段才被保留
minLineLength	int	线段最小长度，小于该值的线段将被忽略
maxLineGap	int	同一线段上点之间的最大间隙

方法列表（常用方法）

方法名	功能
detect(InputArray src, OutputArray lines, Stream& stream = Stream::Null())	执行霍夫变换检测线段
setRho(float rho) / getRho()	设置/获取距离分辨率
setTheta(float theta) / getTheta()	设置/获取角度分辨率
setThreshold(int threshold) / getThreshold()	设置/获取投票阈值
setMinLineLength(int minLineLength) / getMinLineLength()	设置/获取线段最小长度
setMaxLineGap(int maxLineGap) / getMaxLineGap()	设置/获取线段最大间隙

代码示例

#
include <opencv2/opencv.hpp>
  #
  include <opencv2/cudaimgproc.hpp>
    int main(
    )
    {
    // Step 1: 加载图像并转为灰度图
    cv::Mat h_src = cv::imread("/media/dingxin/data/study/OpenCV/sources/images/lines.png"
    , cv::IMREAD_GRAYSCALE)
    ;
    if (h_src.empty(
    )
    ) {
    std::cerr <<
    "无法加载图像！" << std::endl;
    return -1
    ;
    }
    // Step 1.1: 图像预处理（高斯模糊降噪）
    cv::Mat h_blur;
    cv::GaussianBlur(h_src, h_blur, cv::Size(5
    , 5
    )
    , 0
    )
    ;
    // Step 2: 上传到 GPU 并执行 Canny 边缘检测
    cv::cuda::GpuMat d_src, d_edges;
    d_src.upload(h_blur)
    ;
    cv::Ptr<cv::cuda::CannyEdgeDetector> canny = cv::cuda::createCannyEdgeDetector(100
      , 200
      )
      ;
      canny->
      detect(d_src, d_edges)
      ;
      // Step 3: 创建概率霍夫变换检测器
      float rho = 1.0f
      ;
      // 距离分辨率
      float theta = CV_PI / 180.0f
      ;
      // 角度分辨率（1 度）
      int threshold = 30
      ;
      // 投票阈值（更敏感）
      int minLineLength = 50
      ;
      // 最小线段长度（更短也保留）
      int maxLineGap = 20
      ;
      // 线段之间最大间隙（容许更大间隙）
      cv::Ptr<cv::cuda::HoughSegmentDetector> hough =
        cv::cuda::createHoughSegmentDetector(rho, theta, threshold, minLineLength, maxLineGap)
        ;
        // Step 4: 执行线段检测
        cv::cuda::GpuMat d_lines;
        hough->
        detect(d_edges, d_lines)
        ;
        // Step 5: 下载结果
        std::vector<cv::Vec4i> h_lines;
          d_lines.download(h_lines)
          ;
          // Step 6: 绘制检测到的线段
          cv::Mat display;
          cv::cvtColor(h_src, display, cv::COLOR_GRAY2BGR)
          ;
          for (
          const
          auto& line : h_lines)
          {
          cv::Point pt1(line[0]
          , line[1]
          )
          ;
          cv::Point pt2(line[2]
          , line[3]
          )
          ;
          cv::line(display, pt1, pt2, cv::Scalar(0
          , 255
          , 0
          )
          , 2
          )
          ;
          }
          // Step 7: 显示结果
          cv::imshow("Detected Line Segments"
          , display)
          ;
          cv::waitKey(
          )
          ;
          return 0
          ;
          }

运行结果