Modified the thickness of the gaze line

PINTO0309 · Dec 15, 2024 · 1c5a570 · 1c5a570
1 parent a34c1cc
commit 1c5a570
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
@@ -1419,7 +1419,7 @@ def calculate_centroid(heatmap: np.ndarray) -> Tuple[int, int, float]:
         for head_box, heatmap in zip(head_boxes, heatmaps):
             cx, cy, score = calculate_centroid(heatmap)
             if score >= 0.50:
-                cv2.line(debug_image, (head_box.cx, head_box.cy), (cx, cy), (255, 255, 255), thickness=2, lineType=cv2.LINE_AA)
+                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)
                 cv2.circle(debug_image, (cx, cy), 3, (0, 0, 255), thickness=-1, lineType=cv2.LINE_AA)