comment

PINTO0309 · Dec 15, 2024 · 9f9f707 · 9f9f707
1 parent 7e709b9
commit 9f9f707
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/462_Gaze-LLE/demo/demo_yolov9_onnx_gazelle.py b/462_Gaze-LLE/demo/demo_yolov9_onnx_gazelle.py
@@ -1415,6 +1415,7 @@ def calculate_centroid(heatmap: np.ndarray) -> Tuple[int, int, float]:
             y, x = np.unravel_index(max_index, heatmap.shape)
             return int(x), int(y), heatmap[y, x]
 
+        # 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: