From d63f9dcff306a3b66a2b7d9ac8b2f7ed8a75ee0f Mon Sep 17 00:00:00 2001 From: Stuart Date: Fri, 23 Aug 2024 15:01:07 -0400 Subject: [PATCH] Fix ultralytics extract track IDs --- fiftyone/utils/ultralytics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fiftyone/utils/ultralytics.py b/fiftyone/utils/ultralytics.py index 276dc7d038..c5473a669d 100644 --- a/fiftyone/utils/ultralytics.py +++ b/fiftyone/utils/ultralytics.py @@ -58,7 +58,7 @@ def _extract_track_ids(result): return ( result.boxes.id.detach().cpu().numpy().astype(int) if result.boxes.is_track - else [None] * len(result.boxes.conf.size(0)) + else [None] * result.boxes.conf.size(0) )