从OpenCV坎尼边缘检测器获取角度

Get angle from OpenCV Canny edge detector

本文关键字:检测器 获取 边缘检测 边缘 OpenCV      更新时间:2023-10-16

我想使用OpenCV的Canny边缘检测器,如本问题中所述。例如:

cv::Canny(image,contours,10,350); 

但是,我不仅希望获得最终的阈值图像,而且还希望获得每个像素处检测到的边缘角度。这在OpenCV中可能吗?

>canny没有直接给你这个。但是,您可以从 Sobel 变换计算角度,该变换在 canny() 内部使用。

伪代码:

    cv::Canny(image,contours,10,350);
    cv::Sobel(image, dx, CV_64F, 1, 0, 3, 1, 0, cv::BORDER_REPLICATE);
    cv::Sobel(image, dy, CV_64F, 0, 1, 3, 1, 0, cv::BORDER_REPLICATE);
    cv::Mat angle(image.size(), CV_64F)
    foreach (i,j) such that contours[i, j] > 0
    {
        angle[i, j] = atan2(dy[i,j], dx[i , j])
    }

除了使用 for 循环之外,您还可以向返回角度方向灰度图像的函数提供dxdy渐变phase然后将其传递给applyColorMap函数,然后用边缘遮罩它,因此背景是黑色的。

以下是工作流程:

  1. 获取角度

    Mat angles;
    phase(dx, dy, angles, true);
    

    true参数表明角度以度为单位返回。

  2. 将角度范围更改为 0-255,以便您可以转换为CV_8U而不会丢失数据

    angles = angles / 360 * 255;
    

    请注意,angles 仍然是CV_64F类型,因为它来自 Sobel 函数

  3. 转换为CV_8U

    angles.convertTo(angles, CV_8U);
    
  4. 应用您选择的色彩图

    applyColorMap(angles, angles, COLORMAP_HSV);
    

    在这种情况下,我选择HSV颜色图。有关详细信息,请参阅以下内容:https://www.learnopencv.com/applycolormap-for-pseudocoloring-in-opencv-c-python/

  5. 应用边缘蒙版,使背景为黑色

    Mat colored;
    angles.copyTo(colored, contours);
    
  6. 最后显示图像:D

    imshow("Colored angles", colored);
    

如果您的源是视频或网络摄像头,则在应用边缘遮罩之前,您还必须清除colored图像,以防止聚合:

colored.release();
angles.copyTo(colored, contours);

完整代码在这里:

Mat angles, colored;
phase(dx, dy, angles, true);
angles = angles / 360 * 255;
angles.convertTo(angles, CV_8U);
applyColorMap(angles, angles, COLORMAP_HSV);
colored.release();
angles.copyTo(colored, contours);
imshow("Colored angles", colored);