centroid socre threshold 0.50 -> 0.30

PINTO0309 · Dec 16, 2024 · d863f67 · d863f67
1 parent 3b412eb
commit d863f67
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/462_Gaze-LLE/demo/demo_yolov9_onnx_gazelle.py b/462_Gaze-LLE/demo/demo_yolov9_onnx_gazelle.py
@@ -1406,7 +1406,7 @@ def calculate_centroid(heatmap: np.ndarray) -> Tuple[int, int, float]:
         # Drawing of the point of gaze
         for head_box, heatmap in zip(head_boxes, heatmaps):
             cx, cy, score = calculate_centroid(heatmap)
-            if score >= 0.50:
+            if score >= 0.30:
                 cv2.line(debug_image, (head_box.cx, head_box.cy), (cx, cy), (255, 255, 255), thickness=3, lineType=cv2.LINE_AA)
                 cv2.line(debug_image, (head_box.cx, head_box.cy), (cx, cy), (0, 255, 0), thickness=2, lineType=cv2.LINE_AA)
                 cv2.circle(debug_image, (cx, cy), 4, (255, 255, 255), thickness=-1, lineType=cv2.LINE_AA)