diff --git a/supervision/detection/core.py b/supervision/detection/core.py index 825d00c7..392905cd 100644 --- a/supervision/detection/core.py +++ b/supervision/detection/core.py @@ -236,6 +236,9 @@ class Detections: confidence=ultralytics_results.boxes.conf.cpu().numpy(), class_id=ultralytics_results.boxes.cls.cpu().numpy().astype(int), mask=extract_ultralytics_masks(ultralytics_results), + tracker_id=ultralytics_results.boxes.id.int().cpu().numpy() + if ultralytics_results.boxes.id is not None + else None, ) @classmethod