From 76cdadfad5fddb24b2f5152c184a55c80a812c48 Mon Sep 17 00:00:00 2001 From: voxel51-bot <49659360+voxel51-bot@users.noreply.github.com> Date: Fri, 23 Aug 2024 22:31:48 -0400 Subject: [PATCH] Fix ultralytics extract track IDs (#4720) (#4722) Co-authored-by: Stuart --- 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) )