diff --git a/yolox/tracker/byte_tracker.py b/yolox/tracker/byte_tracker.py index 2d004599..f47a3387 100644 --- a/yolox/tracker/byte_tracker.py +++ b/yolox/tracker/byte_tracker.py @@ -158,7 +158,7 @@ def __init__(self, args, frame_rate=30): def update(self, output_results, img_info, img_size): self.frame_id += 1 - activated_starcks = [] + activated_stracks = [] refind_stracks = [] lost_stracks = [] removed_stracks = [] @@ -214,7 +214,7 @@ def update(self, output_results, img_info, img_size): det = detections[idet] if track.state == TrackState.Tracked: track.update(detections[idet], self.frame_id) - activated_starcks.append(track) + activated_stracks.append(track) else: track.re_activate(det, self.frame_id, new_id=False) refind_stracks.append(track) @@ -235,7 +235,7 @@ def update(self, output_results, img_info, img_size): det = detections_second[idet] if track.state == TrackState.Tracked: track.update(det, self.frame_id) - activated_starcks.append(track) + activated_stracks.append(track) else: track.re_activate(det, self.frame_id, new_id=False) refind_stracks.append(track) @@ -254,7 +254,7 @@ def update(self, output_results, img_info, img_size): matches, u_unconfirmed, u_detection = matching.linear_assignment(dists, thresh=0.7) for itracked, idet in matches: unconfirmed[itracked].update(detections[idet], self.frame_id) - activated_starcks.append(unconfirmed[itracked]) + activated_stracks.append(unconfirmed[itracked]) for it in u_unconfirmed: track = unconfirmed[it] track.mark_removed() @@ -266,7 +266,7 @@ def update(self, output_results, img_info, img_size): if track.score < self.det_thresh: continue track.activate(self.kalman_filter, self.frame_id) - activated_starcks.append(track) + activated_stracks.append(track) """ Step 5: Update state""" for track in self.lost_stracks: if self.frame_id - track.end_frame > self.max_time_lost: @@ -276,7 +276,7 @@ def update(self, output_results, img_info, img_size): # print('Ramained match {} s'.format(t4-t3)) self.tracked_stracks = [t for t in self.tracked_stracks if t.state == TrackState.Tracked] - self.tracked_stracks = joint_stracks(self.tracked_stracks, activated_starcks) + self.tracked_stracks = joint_stracks(self.tracked_stracks, activated_stracks) self.tracked_stracks = joint_stracks(self.tracked_stracks, refind_stracks) self.lost_stracks = sub_stracks(self.lost_stracks, self.tracked_stracks) self.lost_stracks.extend(lost_stracks)