You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.4 KiB
88 lines
2.4 KiB
#include <opencv2/line_descriptor.hpp> |
|
|
|
#include "opencv2/core/utility.hpp" |
|
#include "opencv2/core/private.hpp" |
|
#include <opencv2/imgproc.hpp> |
|
#include <opencv2/features2d.hpp> |
|
#include <opencv2/highgui.hpp> |
|
|
|
#include <iostream> |
|
|
|
using namespace cv; |
|
using namespace std; |
|
|
|
static const char* keys = |
|
{ |
|
"{@image_path | | Image path }" |
|
}; |
|
|
|
|
|
static void help() |
|
{ |
|
cout << "\nThis example shows the functionalities of lines extraction " << |
|
"furnished by BinaryDescriptor class\n" << |
|
"Please, run this sample using a command in the form\n" << |
|
"./example_line_descriptor_lines_extraction <path_to_input_image>" << endl; |
|
} |
|
|
|
int main( int argc, char** argv ) |
|
{ |
|
/* get parameters from comand line */ |
|
CommandLineParser parser( argc, argv, keys ); |
|
String image_path = parser.get<String>( 0 ); |
|
|
|
if(image_path.empty()) |
|
{ |
|
help(); |
|
return -1; |
|
} |
|
|
|
/* load image */ |
|
cv::Mat imageMat = imread(image_path, 0); |
|
if(imageMat.data == NULL) |
|
{ |
|
std::cout << "Error, image could not be loaded. Please, check its path" << std::endl; |
|
} |
|
|
|
/* create a ramdom binary mask */ |
|
// cv::Mat mask(imageMat.size(), CV_8UC1); |
|
// cv::randu(mask, Scalar::all(0), Scalar::all(1)); |
|
cv::Mat mask = Mat::ones(imageMat.size(), CV_8UC1); |
|
|
|
/* create a pointer to a BinaryDescriptor object with deafult parameters */ |
|
Ptr<BinaryDescriptor> bd = BinaryDescriptor::createBinaryDescriptor(); |
|
|
|
/* create a structure to store extracted lines */ |
|
vector<KeyLine> lines; |
|
|
|
/* extract lines */ |
|
bd->detect(imageMat, lines, mask); |
|
std::cout << lines.size() << std::endl; |
|
|
|
/* draw lines extracted from octave 0 */ |
|
cv::Mat output = imageMat.clone(); |
|
cvtColor(output, output, COLOR_GRAY2BGR); |
|
for(size_t i = 0; i<lines.size(); i++) |
|
{ |
|
KeyLine kl = lines[i]; |
|
if(kl.octave == 0) |
|
{ |
|
/* get a random color */ |
|
int R = (rand() % (int)(255 + 1)); |
|
int G = (rand() % (int)(255 + 1)); |
|
int B = (rand() % (int)(255 + 1)); |
|
|
|
/* get extremes of line */ |
|
Point pt1 = Point(kl.startPointX, kl.startPointY); |
|
Point pt2 = Point(kl.endPointX, kl.endPointY); |
|
|
|
/* draw line */ |
|
line(output, pt1, pt2, Scalar(B,G,R), 5); |
|
} |
|
|
|
} |
|
|
|
/* show lines on image */ |
|
imshow("Lines", output); |
|
waitKey(); |
|
}
|
|
|