Merge pull request #280 from PJarbas/patch-1

Update demo.py
tryolabs · Jan 30, 2024 · 24ec918 · 24ec918
2 parents 5cc0dd6 + 0b2b420
commit 24ec918
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/demos/yolov5/src/demo.py b/demos/yolov5/src/demo.py
@@ -151,9 +151,7 @@ def yolo_detections_to_norfair_detections(
         )
         tracked_objects = tracker.update(detections=detections)
         if args.track_points == "centroid":
-            norfair.draw_points(frame, detections)
-            norfair.draw_tracked_objects(frame, tracked_objects)
+            norfair.draw_points(frame, tracked_objects)
         elif args.track_points == "bbox":
-            norfair.draw_boxes(frame, detections)
-            norfair.draw_tracked_boxes(frame, tracked_objects)
+            norfair.draw_boxes(frame, tracked_objects)
         video.write(frame)