fix: frame_id in update (#9)

artefactory · Nov 24, 2023 · ac36ee3 · ac36ee3
2 parents 49396da + 1aaeb78
commit ac36ee3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bytetracker/byte_tracker.py b/bytetracker/byte_tracker.py
@@ -180,8 +180,8 @@ def __init__(self, track_thresh=0.45, track_buffer=25, match_thresh=0.8, frame_r
         self.max_time_lost = self.buffer_size
         self.kalman_filter = KalmanFilter()
 
-    def update(self, dets, _):
-        self.frame_id += 1
+    def update(self, dets, frame_id):
+        self.frame_id = frame_id
         activated_starcks = []
         refind_stracks = []
         lost_stracks = []