ultralytics 8.0.56 PyTorch 2.0 support and minor fixes (#1538)

Co-authored-by: N-Friederich <127681326+N-Friederich@users.noreply.github.com>
Co-authored-by: Uhrendoktor <36703334+Uhrendoktor@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: Aman Agarwal <amanag.11@gmail.com>
Co-authored-by: ExtReMLapin <3909752+ExtReMLapin@users.noreply.github.com>
Co-authored-by: Nadav Eidelstein <30617226+nodeav@users.noreply.github.com>
This commit is contained in:
Glenn Jocher
2023-03-24 00:22:20 +01:00
committed by GitHub
parent 8c8c43922c
commit 28e48be5b6
18 changed files with 149 additions and 88 deletions

View File

@ -53,3 +53,7 @@ class BaseTrack:
def mark_removed(self):
self.state = TrackState.Removed
@staticmethod
def reset_id():
BaseTrack._count = 0

View File

@ -168,6 +168,7 @@ class BYTETracker:
self.args = args
self.max_time_lost = int(frame_rate / 30.0 * args.track_buffer)
self.kalman_filter = self.get_kalmanfilter()
self.reset_id()
def update(self, results, img=None):
self.frame_id += 1
@ -299,6 +300,9 @@ class BYTETracker:
def multi_predict(self, tracks):
STrack.multi_predict(tracks)
def reset_id(self):
STrack.reset_id()
@staticmethod
def joint_stracks(tlista, tlistb):
exists = {}