diff --git a/demos/yolov5/src/demo.py b/demos/yolov5/src/demo.py index f2cab00b..00368868 100644 --- 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)