Fix prob (apply wg-perception#101)

sbgisen · May 8, 2024 · ea796ab · ea796ab
1 parent a8ebdf2
commit ea796ab
Showing 1 changed file with 2 additions and 2 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 -
+      float probability = 0.5 - 0.5 *
                           forest->predict(tmp_mat, cv::noArray(), cv::ml::RTrees::PREDICT_SUM) /
-                          forest->getRoots().size();
+                          static_cast<float>(forest->getRoots().size());
 
       tf::Stamped<tf::Point> loc((*i)->center(), scan->header.stamp, scan->header.frame_id);
       try