Attribute bug fixes

Scott Klum
1 parent 2478d1bb
Showing 2 changed files with 5 additions and 3 deletions
openbr/plugins/filter.cpp
openbr/plugins/svm.cpp
@@ -45,7 +45,8 @@ class GammaTransform : public UntrainableTransform
     void project(const Template &src, Template &dst) const
     {
-        LUT(src, lut, dst);
+        if (src.m().depth() == CV_8U) LUT(src, lut, dst);
+        else                          pow(src, gamma, dst);
     }
 };
@@ -74,7 +74,7 @@ static void trainSVM(SVM &amp;svm, Mat data, Mat lab, int kernel, int type, float C,
             svm.train_auto(data, lab, Mat(), Mat(), params, 5);
         } catch (...) {
             qWarning("Some classes do not contain sufficient examples or are not discriminative enough for accurate SVM classification.");
-            svm.train(data, lab);
+            svm.train(data, lab, Mat(), Mat(), params);
         }
     } else {
         params.C = C;
@@ -161,7 +161,8 @@ private:
             dst.m().at<float>(0, 0) = prediction;
             // positive values ==> first class
             // negative values ==> second class
-            prediction = prediction > 0 ? 0 : 1;
+            if (type != EPS_SVR && type != NU_SVR)
+                prediction = prediction > 0 ? 0 : 1;
         }
         if (type == EPS_SVR || type == NU_SVR) {
             dst.file.set(outputVariable, prediction);