Merge pull request #795 from sovrasov:tld_grayscale

23bda58f · Alexander Alekhin · 2283cea6 · f857b7f7 · 23bda58f
Commit 23bda58f authored Oct 06, 2016 by Alexander Alekhin
Hide whitespace changes
Inline Side-by-side

Showing with 9 additions and 1 deletion

tldTracker.cpp modules/tracking/src/tldTracker.cpp +9 -1

No files found.
--- 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;