Adding svm regressor output to dst template data

Brendan Klare
1 parent f239ab13
Showing 1 changed file with 5 additions and 2 deletions
openbr/plugins/svm.cpp
@@ -163,9 +163,12 @@ private:
             // negative values ==> second class
             prediction = prediction > 0 ? 0 : 1;
         }
-        if (type == EPS_SVR || type == NU_SVR)
+        if (type == EPS_SVR || type == NU_SVR) {
             dst.file.set(outputVariable, prediction);
-        else
+            dst.m() = Mat(1, 1, CV_32F);
+            dst.m().at<float>(0, 0) = prediction;
+
+        } else
             dst.file.set(outputVariable, reverseLookup[prediction]);
     }