opencv/samples/cpp/lsd_lines.cpp

55 lines
1.3 KiB
C++
Raw Normal View History

#include <iostream>
#include <string>
#include "opencv2/core/core.hpp"
#include "opencv2/core/utility.hpp"
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/highgui/highgui.hpp"
using namespace std;
using namespace cv;
int main(int argc, char** argv)
{
2013-07-22 12:49:33 +02:00
std::string in;
if (argc != 2)
{
2013-07-22 12:49:33 +02:00
std::cout << "Usage: lsd_lines [input image]. Now loading building.jpg" << std::endl;
in = "building.jpg";
}
else
{
in = argv[1];
}
Mat image = imread(in, IMREAD_GRAYSCALE);
2013-07-22 12:49:33 +02:00
#if 0
Canny(image, image, 50, 200, 3); // Apply canny edge
#endif
// Create and LSD detector with standard or no refinement.
#if 1
Ptr<LineSegmentDetector> ls = createLineSegmentDetectorPtr(LSD_REFINE_STD);
#else
Ptr<LineSegmentDetector> ls = createLineSegmentDetectorPtr(LSD_REFINE_NONE);
#endif
double start = double(getTickCount());
vector<Vec4i> lines_std;
2013-07-22 12:49:33 +02:00
// Detect the lines
ls->detect(image, lines_std);
2013-07-22 12:49:33 +02:00
double duration_ms = (double(getTickCount()) - start) * 1000 / getTickFrequency();
std::cout << "It took " << duration_ms << " ms." << std::endl;
2013-07-22 12:49:33 +02:00
// Show found lines
Mat drawnLines(image);
2013-07-22 13:46:34 +02:00
lsd_std->drawSegments(drawnLines, lines_std);
imshow("Standard refinement", drawnLines);
waitKey();
return 0;
}