From a2aca62ecb19397ed141d0399e3f8778907d2bd3 Mon Sep 17 00:00:00 2001 From: Raif Olson Date: Thu, 28 Mar 2024 09:27:54 -0400 Subject: [PATCH] merge develop branch and change back to using boolean indexing --- supervision/tracker/byte_tracker/core.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/supervision/tracker/byte_tracker/core.py b/supervision/tracker/byte_tracker/core.py index eb94e55a..c77878cf 100644 --- a/supervision/tracker/byte_tracker/core.py +++ b/supervision/tracker/byte_tracker/core.py @@ -287,13 +287,7 @@ class ByteTrack: for i_detection, i_track in matches: detections.tracker_id[i_detection] = int(tracks[i_track].track_id) - invalid_tracks = [ - i - for i in range(len(detections.tracker_id)) - if detections.tracker_id[i] != -1 - ] - - return detections[invalid_tracks] + return detections[detections.tracker_id != -1] else: detections.tracker_id = np.array([], dtype=int)