samples/cpp/lsd_lines.cpp

使用 LineSegmentDetector 的示例

示例输出图像

#include "opencv2/imgproc.hpp"
#include "opencv2/imgcodecs.hpp"
#include "opencv2/highgui.hpp"
#include <iostream>
 
using namespace std;
using namespace cv;
 
int main(int argc, char** argv)
{
    cv::CommandLineParser parser(argc, argv,
                                 "{input i|building.jpg|输入图像}"
                                 "{refine r|false|如果为 true，则使用 LSD_REFINE_STD 方法，如果为 false，则使用 LSD_REFINE_NONE 方法}"
                                 "{canny c|false|使用 Canny 边缘检测器}"
                                 "{overlay o|false|在输入图像上显示结果}"
                                 "{help h|false|显示帮助信息}");
 
    if (parser.get<bool>("help"))
    {
parser.printMessage();
        return 0;
    }
 
parser.printMessage();
 
    String filename = samples::findFile(parser.get<String>("input"));
    bool useRefine = parser.get<bool>("refine");
    bool useCanny = parser.get<bool>("canny");
    bool overlay = parser.get<bool>("overlay");
 
    Mat image = imread(filename, IMREAD_GRAYSCALE);
 
    if( image.empty() )
    {
cout << "无法加载 " << filename;
        return 1;
    }
 
imshow("源图像", image);
 
    if (useCanny)
    {
Canny(image, image, 50, 200, 3); // 应用 Canny 边缘检测器
    }
 
    // 创建并使用标准或无细化的方法创建 LSD 检测器。
    Ptr<LineSegmentDetector> ls = useRefine ? createLineSegmentDetector(LSD_REFINE_STD) : createLineSegmentDetector(LSD_REFINE_NONE);
 
    double start = double(getTickCount());
vector<Vec4f> lines_std;
 
    // 检测线条
ls->detect(image, lines_std);
 
    double duration_ms = (double(getTickCount()) - start) * 1000 / getTickFrequency();
std::cout << "耗时 " << duration_ms << " 毫秒。" << std::endl;
 
    // 显示找到的线
    if (!overlay || useCanny)
    {
image = Scalar(0, 0, 0);
    }
 
ls->drawSegments(image, lines_std);
 
    String window_name = useRefine ? "结果 - 标准细化" : "结果 - 无细化";
window_name += useCanny ? " - 使用 Canny 边缘检测器" : "";
 
imshow(window_name, image);
 
waitKey();
    return 0;
}