From 92bb32b816d955a93886260073ce7e39184be340 Mon Sep 17 00:00:00 2001 From: Nicolas Date: Wed, 27 Mar 2024 17:16:56 +0100 Subject: [PATCH] cleaned dead code on byte_tracker --- bytetracker/byte_tracker.py | 21 +-------------------- 1 file changed, 1 insertion(+), 20 deletions(-) diff --git a/bytetracker/byte_tracker.py b/bytetracker/byte_tracker.py index 26c323e..0768efb 100644 --- a/bytetracker/byte_tracker.py +++ b/bytetracker/byte_tracker.py @@ -36,7 +36,6 @@ def __init__(self, tlwh, score, cls): self.is_activated = False self.score = score - self.tracklet_len = 0 self.cls = cls def predict(self): @@ -66,7 +65,6 @@ def activate(self, kalman_filter, frame_id): self.track_id = self.next_id() self.mean, self.covariance = self.kalman_filter.initiate(self.tlwh_to_xyah(self._tlwh)) - self.tracklet_len = 0 self.state = TrackState.Tracked if frame_id == 1: self.is_activated = True @@ -78,7 +76,6 @@ def re_activate(self, new_track, frame_id, new_id=False): self.mean, self.covariance = self.kalman_filter.update( self.mean, self.covariance, self.tlwh_to_xyah(new_track.tlwh) ) - self.tracklet_len = 0 self.state = TrackState.Tracked self.is_activated = True self.frame_id = frame_id @@ -96,7 +93,6 @@ def update(self, new_track, frame_id): :return: """ self.frame_id = frame_id - self.tracklet_len += 1 self.cls = new_track.cls new_tlwh = new_track.tlwh @@ -145,13 +141,6 @@ def tlwh_to_xyah(tlwh): def to_xyah(self): return self.tlwh_to_xyah(self.tlwh) - @staticmethod - # @jit(nopython=True) - def tlbr_to_tlwh(tlbr): - ret = np.asarray(tlbr).copy() - ret[2:] -= ret[:2] - return ret - @staticmethod # @jit(nopython=True) def tlwh_to_tlbr(tlwh): @@ -180,14 +169,6 @@ 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 reset(self): - self.tracked_stracks = [] # type: list[STrack] - self.lost_stracks = [] # type: list[STrack] - self.removed_stracks = [] # type: list[STrack] - self.frame_id = 0 - self.kalman_filter = KalmanFilter() - BaseTrack._count = 0 - def update(self, dets, frame_id): self.frame_id = frame_id activated_starcks = [] @@ -262,7 +243,7 @@ def update(self, dets, frame_id): strack_pool[i] for i in u_track if strack_pool[i].state == TrackState.Tracked ] dists = matching.iou_distance(r_tracked_stracks, detections_second) - matches, u_track, u_detection_second = matching.linear_assignment(dists, thresh=0.5) + matches, u_track, _ = matching.linear_assignment(dists, thresh=0.5) for itracked, idet in matches: track = r_tracked_stracks[itracked] det = detections_second[idet]