Skip to content

Commit

Permalink
cleaned dead code on byte_tracker
Browse files Browse the repository at this point in the history
  • Loading branch information
nmathieufact committed Mar 27, 2024
1 parent 5baeeb5 commit 92bb32b
Showing 1 changed file with 1 addition and 20 deletions.
21 changes: 1 addition & 20 deletions bytetracker/byte_tracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down Expand Up @@ -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
Expand All @@ -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
Expand All @@ -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
Expand Down Expand Up @@ -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):
Expand Down Expand Up @@ -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 = []
Expand Down Expand Up @@ -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]
Expand Down

0 comments on commit 92bb32b

Please sign in to comment.