mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-15 22:20:58 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update lsd.cpp

Browse Source
This commit is contained in:
Suleyman TURKMEN 2017-11-11 11:59:55 +03:00
parent 2e772510ea
commit 992a81dcaa
1 changed files with 2 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
modules/imgproc/src/lsd.cpp
View File

@ -1146,23 +1146,10 @@ void LineSegmentDetectorImpl::drawSegments(InputOutputArray _image, InputArray l
CV_Assert(!_image.empty() && (_image.channels() == 1 || _image.channels() == 3)); CV_Assert(!_image.empty() && (_image.channels() == 1 || _image.channels() == 3));
Mat gray;
if (_image.channels() == 1) if (_image.channels() == 1)
{ {
gray = _image.getMatRef(); cvtColor(_image, _image, COLOR_GRAY2BGR);
} }
else if (_image.channels() == 3)
{
cvtColor(_image, gray, CV_BGR2GRAY);
}
// Create a 3 channel image in order to draw colored lines
std::vector<Mat> planes;
planes.push_back(gray);
planes.push_back(gray);
planes.push_back(gray);
merge(planes, _image);
Mat _lines; Mat _lines;
_lines = lines.getMat(); _lines = lines.getMat();
@ -1174,7 +1161,7 @@ void LineSegmentDetectorImpl::drawSegments(InputOutputArray _image, InputArray l
const Vec4f& v = _lines.at<Vec4f>(i); const Vec4f& v = _lines.at<Vec4f>(i);
Point2f b(v[0], v[1]); Point2f b(v[0], v[1]);
Point2f e(v[2], v[3]); Point2f e(v[2], v[3]);
line(_image.getMatRef(), b, e, Scalar(0, 0, 255), 1); line(_image, b, e, Scalar(0, 0, 255), 1);
} }
} }