Merge branch 'master' of https://github.com/mikel-brostrom/yolo_tracking

README.md

@@ -293,16 +293,17 @@ while True:
     xyxys = tracks[:, 0:4].astype('int') # float64 to int
     ids = tracks[:, 4].astype('int') # float64 to int
     confs = tracks[:, 5]
-    clss = tracks[:, 6]
-    ind = tracks[:, 7]
+    clss = tracks[:, 6].astype('int') # float64 to int
+    inds = tracks[:, 7].astype('int') # float64 to int
 
     # in case you have segmentations or poses alongside with your detections you can use
     # the ind variable in order to identify which track is associated to each seg or pose by:
-    # segs = segs[ind]
-    # poses = poses[ind]
+    # segs = segs[inds]
+    # poses = poses[inds]
+    # you can then zip them together: zip(tracks, poses)
 
     # print bboxes with their associated id, cls and conf
-    if ts.shape[0] != 0:
+    if tracks.shape[0] != 0:
         for xyxy, id, conf, cls in zip(xyxys, ids, confs, clss):
             im = cv2.rectangle(
                 im,

boxmot/__init__.py

@@ -1,6 +1,6 @@
 # Mikel Broström 🔥 Yolo Tracking 🧾 AGPL-3.0 license
 
-__version__ = '10.0.34'
+__version__ = '10.0.35'
 
 from boxmot.postprocessing.gsi import gsi
 from boxmot.tracker_zoo import create_tracker, get_tracker_config

examples/track.py

@@ -13,7 +13,7 @@
 from examples.detectors import get_yolo_inferer
 
 __tr = TestRequirements()
-__tr.check_packages(('ultralytics @ git+https://github.com/mikel-brostrom/ultralytics.git@det-ind', ))  # install
+__tr.check_packages(('ultralytics @ git+https://github.com/mikel-brostrom/ultralytics.git', ))  # install
 
 from ultralytics import YOLO
 from ultralytics.data.utils import VID_FORMATS

examples/val.py

@@ -23,7 +23,7 @@
 from boxmot.utils.checks import TestRequirements
 
 __tr = TestRequirements()
-__tr.check_packages(('ultralytics @ git+https://github.com/mikel-brostrom/ultralytics.git@det-ind', ))  # install
+__tr.check_packages(('ultralytics @ git+https://github.com/mikel-brostrom/ultralytics.git', ))  # install
 
 import git
 from git import Repo