diff --git a/yolox/tracker/byte_tracker.py b/yolox/tracker/byte_tracker.py index 2d004599..6aed9b71 100644 --- a/yolox/tracker/byte_tracker.py +++ b/yolox/tracker/byte_tracker.py @@ -7,7 +7,7 @@ import torch.nn.functional as F from .kalman_filter import KalmanFilter -from yolox.tracker import matching +from ByteTrack.yolox.tracker import matching from .basetrack import BaseTrack, TrackState class STrack(BaseTrack): @@ -15,7 +15,7 @@ class STrack(BaseTrack): def __init__(self, tlwh, score): # wait activate - self._tlwh = np.asarray(tlwh, dtype=np.float) + self._tlwh = np.asarray(tlwh, dtype=float) self.kalman_filter = None self.mean, self.covariance = None, None self.is_activated = False