[melodic] Fix leg probabilities (wg-perception#100)

s-kodama-isp · Jul 2, 2021 · a02dc95 · a02dc95
1 parent bc73fa0
commit a02dc95
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
@@ -732,9 +732,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