ultralytics tracker support added

This commit is contained in:
Hardik Dava committed 2023-08-07 21:19:29 +02:00
1 parent 525e1b00ec
commit 2166bc23bc
1 file changed
+3
+3
View File
@@ -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