Convert Mat if invalid type for hist

Austin Van Blanton
1 parent dfc7cd72
Showing 1 changed file with 6 additions and 2 deletions
openbr/plugins/hist.cpp
@@ -52,8 +52,12 @@ class HistTransform : public UntrainableTransform
             int histSize[] = {dims};
             float range[] = {min, max};
             const float* ranges[] = {range};
-            Mat hist;
-            calcHist(&mv[i], 1, channels, Mat(), hist, 1, histSize, ranges);
+            Mat hist, chan = mv[i];
+            // calcHist requires F or U, might as well convert just in case
+            if (mv[i].depth() != CV_8U || mv[i].depth() == CV_32F)
+                mv[i].convertTo(chan, CV_32FC1);
+//            qDebug("clust r %i c %i ch %i", chan.rows, chan.cols, chan.channels());
+            calcHist(&chan, 1, channels, Mat(), hist, 1, histSize, ranges);
             memcpy(m.ptr(i), hist.ptr(), dims * sizeof(float));
         }