mirror of https://github.com/opencv/opencv.git
Merge pull request #20797 from sturkmen72:restore_LSD
commit
48c985e775
4 changed files with 1061 additions and 18 deletions
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,73 @@ |
||||
#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|input image}" |
||||
"{refine r|false|if true use LSD_REFINE_STD method, if false use LSD_REFINE_NONE method}" |
||||
"{canny c|false|use Canny edge detector}" |
||||
"{overlay o|false|show result on input image}" |
||||
"{help h|false|show help message}"); |
||||
|
||||
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 << "Unable to load " << filename; |
||||
return 1; |
||||
} |
||||
|
||||
imshow("Source Image", image); |
||||
|
||||
if (useCanny) |
||||
{ |
||||
Canny(image, image, 50, 200, 3); // Apply Canny edge detector
|
||||
} |
||||
|
||||
// Create and LSD detector with standard or no refinement.
|
||||
Ptr<LineSegmentDetector> ls = useRefine ? createLineSegmentDetector(LSD_REFINE_STD) : createLineSegmentDetector(LSD_REFINE_NONE); |
||||
|
||||
double start = double(getTickCount()); |
||||
vector<Vec4f> lines_std; |
||||
|
||||
// Detect the lines
|
||||
ls->detect(image, lines_std); |
||||
|
||||
double duration_ms = (double(getTickCount()) - start) * 1000 / getTickFrequency(); |
||||
std::cout << "It took " << duration_ms << " ms." << std::endl; |
||||
|
||||
// Show found lines
|
||||
if (!overlay || useCanny) |
||||
{ |
||||
image = Scalar(0, 0, 0); |
||||
} |
||||
|
||||
ls->drawSegments(image, lines_std); |
||||
|
||||
String window_name = useRefine ? "Result - standard refinement" : "Result - no refinement"; |
||||
window_name += useCanny ? " - Canny edge detector used" : ""; |
||||
|
||||
imshow(window_name, image); |
||||
|
||||
waitKey(); |
||||
return 0; |
||||
} |
Loading…
Reference in new issue