Fix prob (apply wg-perception#101)

sbgisen · May 9, 2024 · 20f7f3a · 20f7f3a
1 parent a8ebdf2
commit 20f7f3a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/leg_detector/src/leg_detector.cpp b/leg_detector/src/leg_detector.cpp
@@ -731,9 +731,9 @@ class LegDetector
 
       memcpy(tmp_mat.data, f.data(), f.size()*sizeof(float));
 
-      float probability = 0.5 -
-                          forest->predict(tmp_mat, cv::noArray(), cv::ml::RTrees::PREDICT_SUM) /
-                          forest->getRoots().size();
+      float probability = 0.5 + 0.5 *
+                          static_cast<float>(forest->predict(tmp_mat, cv::noArray(), cv::ml::RTrees::PREDICT_SUM)) /
+                          static_cast<float>(forest->getRoots().size());
 
       tf::Stamped<tf::Point> loc((*i)->center(), scan->header.stamp, scan->header.frame_id);
       try