|
|
|
|
|
|
|
|
// One (this) to do all kinds of stuff to the picture (grayscale conversion, threshold, gauss etc etc) |
|
|
// One (this) to do all kinds of stuff to the picture (grayscale conversion, threshold, gauss etc etc) |
|
|
// And one (the other one) to segment the lines. |
|
|
// And one (the other one) to segment the lines. |
|
|
// No return value here as the input is passed by reference -> directly modified. |
|
|
// No return value here as the input is passed by reference -> directly modified. |
|
|
vector<Vec4i> lines; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
vector<Vec4i> lines; |
|
|
|
|
|
|
|
|
cvtColor(inputPicture, inputPicture, COLOR_BGR2GRAY); |
|
|
cvtColor(inputPicture, inputPicture, COLOR_BGR2GRAY); |
|
|
threshold(inputPicture, inputPicture, thresholdValue, 255, THRESH_BINARY); |
|
|
threshold(inputPicture, inputPicture, thresholdValue, 255, THRESH_BINARY); |
|
|
|
|
|
|
|
|
line( inputPicture, Point(lines[i][0], lines[i][1]), |
|
|
line( inputPicture, Point(lines[i][0], lines[i][1]), |
|
|
Point( lines[i][2], lines[i][3]), Scalar(0,0,255), 3, 8 ); |
|
|
Point( lines[i][2], lines[i][3]), Scalar(0,0,255), 3, 8 ); |
|
|
} |
|
|
} |
|
|
|
|
|
vector<VectorOfLines> vectors; |
|
|
|
|
|
|
|
|
|
|
|
Point point11; |
|
|
|
|
|
Point point12; |
|
|
|
|
|
Point point21; |
|
|
|
|
|
Point point22; |
|
|
|
|
|
|
|
|
|
|
|
for( size_t i = 0; i < lines.size(); i++ ) |
|
|
|
|
|
{ |
|
|
|
|
|
point11 = Point(lines[i][0], lines[i][1]); |
|
|
|
|
|
point12 = Point( lines[i][2], lines[i][3]); |
|
|
|
|
|
float gradient1 = VectorOfLines::calcGradient(point11, point12); |
|
|
|
|
|
for( size_t j = 0; j < lines.size(); j++ ) |
|
|
|
|
|
{ |
|
|
|
|
|
if(j != i){ |
|
|
|
|
|
point21 = Point(lines[j][0], lines[j][1]); |
|
|
|
|
|
point22 = Point(lines[j][2], lines[j][3]); |
|
|
|
|
|
float gradient2 = VectorOfLines::calcGradient(point21, point22); |
|
|
|
|
|
float gradient12 = VectorOfLines::calcGradient(point12, point21); |
|
|
|
|
|
if((norm(gradient1 - gradient2) < 0.05) & (norm(gradient1 - gradient12) < 0.05)) |
|
|
|
|
|
{ |
|
|
|
|
|
//To Do: add line between 2 lines |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
} |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
imshow("Result", inputPicture); |
|
|
imshow("Result", inputPicture); |
|
|
waitKey(0); |
|
|
waitKey(0); |