Fine-tuning

PINTO0309 · Dec 15, 2024 · a34c1cc · a34c1cc
1 parent 9f9f707
commit a34c1cc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/462_Gaze-LLE/demo/demo_yolov9_onnx_gazelle.py b/462_Gaze-LLE/demo/demo_yolov9_onnx_gazelle.py
@@ -1421,8 +1421,8 @@ def calculate_centroid(heatmap: np.ndarray) -> Tuple[int, int, float]:
             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), (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)
+                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)
 
         if file_paths is not None:
             basename = os.path.basename(file_paths[file_paths_count])