ultralytics 8.0.203 DDP argparse and Tracker fixes (#6007)

ultralytics/__init__.py

@@ -1,6 +1,6 @@
 # Ultralytics YOLO 🚀, AGPL-3.0 license
 
-__version__ = '8.0.202'
+__version__ = '8.0.203'
 
 from ultralytics.models import RTDETR, SAM, YOLO
 from ultralytics.models.fastsam import FastSAM

ultralytics/trackers/bot_sort.py

@@ -192,3 +192,8 @@ class BOTSORT(BYTETracker):
     def multi_predict(self, tracks):
         """Predict and track multiple objects with YOLOv8 model."""
         BOTrack.multi_predict(tracks)
+
+    def reset(self):
+        """Reset tracker."""
+        super().reset()
+        self.gmc.reset_params()

ultralytics/trackers/byte_tracker.py

@@ -374,6 +374,15 @@ class BYTETracker:
         """Resets the ID counter of STrack."""
         STrack.reset_id()
 
+    def reset(self):
+        """Reset tracker."""
+        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 = self.get_kalmanfilter()
+        self.reset_id()
+
     @staticmethod
     def joint_stracks(tlista, tlistb):
         """Combine two lists of stracks into a single one."""

ultralytics/trackers/track.py

@@ -1,6 +1,7 @@
 # Ultralytics YOLO 🚀, AGPL-3.0 license
 
 from functools import partial
+from pathlib import Path
 
 import torch
 
@@ -40,8 +41,12 @@ def on_predict_start(predictor, persist=False):
 def on_predict_postprocess_end(predictor):
     """Postprocess detected boxes and update with object tracking."""
     bs = predictor.dataset.bs
-    im0s = predictor.batch[1]
+    path, im0s = predictor.batch[:2]
+
     for i in range(bs):
+        if predictor.vid_path[i] != str(predictor.save_dir / Path(path[i]).name):  # new video
+            predictor.trackers[i].reset()
+
         det = predictor.results[i].boxes.cpu().numpy()
         if len(det) == 0:
             continue

ultralytics/trackers/utils/gmc.py

@@ -300,3 +300,10 @@ class GMC:
         self.prevKeyPoints = copy.copy(keypoints)
 
         return H
+
+    def reset_params(self):
+        """Reset parameters."""
+        self.prevFrame = None
+        self.prevKeyPoints = None
+        self.prevDescriptors = None
+        self.initializedFirstFrame = False