Skip to content

Commit 3865e3c

Browse files
committed
Merge pull request #565 from LorenaGdL:MedianFlow_tracker_fix
2 parents bb8d424 + 5b70688 commit 3865e3c

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

modules/tracking/src/trackerMedianFlow.cpp

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -190,8 +190,15 @@ bool TrackerMedianFlowImpl::medianFlowImpl(Mat oldImage,Mat newImage,Rect2d& old
190190
std::vector<Point2f> pointsToTrackOld,pointsToTrackNew;
191191

192192
Mat oldImage_gray,newImage_gray;
193-
cvtColor( oldImage, oldImage_gray, COLOR_BGR2GRAY );
194-
cvtColor( newImage, newImage_gray, COLOR_BGR2GRAY );
193+
if (oldImage.channels() != 1)
194+
cvtColor( oldImage, oldImage_gray, COLOR_BGR2GRAY );
195+
else
196+
oldImage.copyTo(oldImage_gray);
197+
198+
if (newImage.channels() != 1)
199+
cvtColor( newImage, newImage_gray, COLOR_BGR2GRAY );
200+
else
201+
newImage.copyTo(newImage_gray);
195202

196203
//"open ended" grid
197204
for(int i=0;i<params.pointsInGrid;i++){

0 commit comments

Comments
 (0)