bug fixes

Josh Klontz
1 parent c9e43a5f
Showing 5 changed files with 20 additions and 16 deletions
sdk/core/opencvutils.cpp
sdk/plugins/algorithms.cpp
sdk/plugins/cvt.cpp
sdk/plugins/eigen3.cpp
sdk/plugins/integral.cpp
@@ -143,6 +143,7 @@ Mat OpenCVUtils::toMatByRow(const QList&lt;Mat&gt; &amp;src)
     int rows = 0; foreach (const Mat &m, src) rows += m.rows;
     int cols = src.first().cols;
+    if (cols == 0) qFatal("Columnless matrix!");
     int type = src.first().type();
     Mat dst(rows, cols, type);
@@ -50,7 +50,7 @@ class AlgorithmsInitializer : public Initializer
         Globals->abbreviations.insert("SmallSIFT", "Open+LimitSize(512)+KeyPointDetector(SIFT)+KeyPointDescriptor(SIFT):KeyPointMatcher(BruteForce)");
         Globals->abbreviations.insert("SmallSURF", "Open+LimitSize(512)+KeyPointDetector(SURF)+KeyPointDescriptor(SURF):KeyPointMatcher(BruteForce)");
         Globals->abbreviations.insert("ColorHist", "Open+LimitSize(512)!EnsureChannels(3)+SplitChannels+Hist(256,0,8)+Cat+Normalize(L1):L2");
-        Globals->abbreviations.insert("IHH", "Open+(RG+MAdd(0.5))/(Cvt(Gray)+Gradient+Bin(0,360,8,true))+Merge+Integral+IntegralSampler+CvtFloat+WordWise(RowWisePCA(8)+RowWiseMeanCenter+Binarize,Identity):L2");
+        Globals->abbreviations.insert("IHH", "Open+(RG+MAdd(0.5))/(Cvt(Gray)+Gradient+Bin(0,360,8,true))+Merge+Integral+IntegralSampler+CvtFloat+WordWise(RowWisePCA(8)+RowWiseMeanCenter+Binarize,RowWisePCA):L2");
         // Hash
         Globals->abbreviations.insert("FileName", "Name+Identity:Identical");
@@ -181,12 +181,17 @@ class RGTransform : public UntrainableTransform
         for (int i=0; i<m.rows; i++)
             for (int j=0; j<m.cols; j++) {
                 Vec3b v = m.at<Vec3b>(i,j);
-                uchar& b = v[0];
-                uchar& g = v[1];
-                uchar& r = v[2];
-
-                R.at<uchar>(i, j) = saturate_cast<uchar>(255.0*r/(r+g+b));
-                G.at<uchar>(i, j) = saturate_cast<uchar>(255.0*g/(r+g+b));
+                const int b = v[0];
+                const int g = v[1];
+                const int r = v[2];
+                const int sum = b + g + r;
+                if (sum > 0) {
+                    R.at<uchar>(i, j) = saturate_cast<uchar>(255.0*r/(r+g+b));
+                    G.at<uchar>(i, j) = saturate_cast<uchar>(255.0*g/(r+g+b));
+                } else {
+                    R.at<uchar>(i, j) = 0;
+                    G.at<uchar>(i, j) = 0;
+                }
             }
         dst.append(R);
@@ -145,15 +145,15 @@ protected:
         if (keep < 1) {
             // Keep eigenvectors that retain a certain energy percentage.
-            double totalEnergy = allEVals.sum();
+            const double totalEnergy = allEVals.sum();
             if (totalEnergy == 0) {
                 keep = 0;
             } else {
                 double currentEnergy = 0;
-                int i;
-                for (i=1; i<=allEVals.rows(); i++) {
-                    currentEnergy += allEVals(allEVals.rows()-i);
-                    if (currentEnergy / totalEnergy >= keep) break;
+                int i=0;
+                while ((currentEnergy / totalEnergy < keep) && (i < allEVals.rows())) {
+                    currentEnergy += allEVals(allEVals.rows()-(i+1));
+                    i++;
                 }
                 keep = i - drop;
             }
@@ -159,9 +159,7 @@ class WordWiseTransform : public Transform
     void project(const Template &src, Template &dst) const
     {
-        Template reworded;
-        getWords->project(src, reworded);
-        byWord->project(reworded, dst);
+        byWord->project(reword(src), dst);
     }
     Template reword(const Template &src) const
@@ -177,7 +175,7 @@ class WordWiseTransform : public Transform
         QVector<int> indicies(numWords, 0);
         for (int i=0; i<src.m().rows; i++) {
             const int word = words.m().at<uchar>(i,0);
-            reworded[word].row(indicies[word]++) = src.m().row(i);
+            src.m().row(i).copyTo(reworded[word].row(indicies[word]++));
         }
         return reworded;
     }