Fix ultralytics extract track IDs (#4720)

voxel51 · Aug 23, 2024 · 2deee54 · 2deee54
1 parent e21f1ba
commit 2deee54
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)
     )