use select_fields() to optimize

voxel51 · May 8, 2024 · 303c416 · 303c416
1 parent 75c9f77
commit 303c416
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/fiftyone/utils/tracking/deepsort.py b/fiftyone/utils/tracking/deepsort.py
@@ -68,9 +68,9 @@ def track(
             sample_collection, _in_field, fo.Detections
         )
 
-        for sample in sample_collection.iter_samples(
-            autosave=True, progress=progress
-        ):
+        view = sample_collection.select_fields(_in_field)
+
+        for sample in view.iter_samples(autosave=True, progress=progress):
             try:
                 DeepSort.track_sample(
                     sample,