Skip to content

Commit

Permalink
reverted one change
Browse files Browse the repository at this point in the history
  • Loading branch information
nmathieufact committed Mar 27, 2024
1 parent 62233ed commit d996bbe
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions bytetracker/byte_tracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -141,6 +141,13 @@ 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 @@ -169,6 +176,14 @@ 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

0 comments on commit d996bbe

Please sign in to comment.