From ccd30468bcde72a63d5ba36e4abf067b140472a8 Mon Sep 17 00:00:00 2001 From: Zhiyuan Chen Date: Sun, 26 Apr 2020 15:03:21 +0800 Subject: [PATCH] Reduce number of loops in tracker Slightly increase spped by reducing two for loop to one --- deep_sort/tracker.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/deep_sort/tracker.py b/deep_sort/tracker.py index de99de44..e7701155 100644 --- a/deep_sort/tracker.py +++ b/deep_sort/tracker.py @@ -103,10 +103,10 @@ def gated_metric(tracks, dets, track_indices, detection_indices): return cost_matrix # Split track set into confirmed and unconfirmed tracks. - confirmed_tracks = [ - i for i, t in enumerate(self.tracks) if t.is_confirmed()] - unconfirmed_tracks = [ - i for i, t in enumerate(self.tracks) if not t.is_confirmed()] + confirmed_tracks = list() + unconfirmed_tracks = list() + for i, t in enumerate(self.tracks): + confirmed_tracks.append(i) if t.is_confirmed() else unconfirmed_tracks.append(i) # Associate confirmed tracks using appearance features. matches_a, unmatched_tracks_a, unmatched_detections = \