diff --git a/deep_sort/track.py b/deep_sort/track.py index f638e9b3e..a1f9c16fe 100644 --- a/deep_sort/track.py +++ b/deep_sort/track.py @@ -72,7 +72,6 @@ def __init__(self, mean, covariance, track_id, n_init, max_age, self.age = 1 self.time_since_update = 0 - self.state = TrackState.Tentative self.features = [] if feature is not None: self.features.append(feature) @@ -80,6 +79,8 @@ def __init__(self, mean, covariance, track_id, n_init, max_age, self._n_init = n_init self._max_age = max_age + self.state = TrackState.Tentative if self.hits < self._n_init else TrackState.Confirmed + def to_tlwh(self): """Get current position in bounding box format `(top left x, top left y, width, height)`.