Fix ultralytics extract track IDs (#4720) (#4722)

Co-authored-by: Stuart <[email protected]>
voxel51 · Aug 24, 2024 · 76cdadf · 76cdadf
1 parent b19a544
commit 76cdadf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)
     )