Fix TLD fail on gray images

9 years ago · f857b7f77c
parent 1df640a6a2
commit f857b7f77c
1 changed files with 9 additions and 1 deletions
--- a/modules/tracking/src/tldTracker.cpp
+++ b/modules/tracking/src/tldTracker.cpp
@ -82,7 +82,15 @@ bool TrackerTLDImpl::initImpl(const Mat& image, const Rect2d& boundingBox)
 {
    Mat image_gray;
    trackerProxy->init(image, boundingBox);
-    cvtColor( image, image_gray, COLOR_BGR2GRAY );
+    if(image.channels() > 1)
+    {
+        cvtColor( image, image_gray, COLOR_BGR2GRAY );
+    }
+    else
+    {
+        image_gray = image.clone();
+    }
+
    data = Ptr<Data>(new Data(boundingBox));
    double scale = data->getScale();
    Rect2d myBoundingBox = boundingBox;