Merge pull request #1089 from mikel-brostrom/hybridsort-per-class

Hybridsort per class
mikel-brostrom · Aug 22, 2023 · f1231ee · f1231ee
2 parents b5b441a + b590ce3
commit f1231ee
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/boxmot/trackers/hybridsort/hybridsort.py b/boxmot/trackers/hybridsort/hybridsort.py
@@ -13,6 +13,7 @@
 from boxmot.trackers.hybridsort.association import (
     associate_4_points_with_score, associate_4_points_with_score_with_reid,
     cal_score_dif_batch_two_score, embedding_distance, linear_assignment)
+from boxmot.utils import PerClassDecorator
 from boxmot.utils.iou import get_asso_func
 
 np.random.seed(0)
@@ -339,6 +340,7 @@ def __init__(self, reid_weights, device, half, det_thresh, max_age=30, min_hits=
         self.min_hits = min_hits
         self.iou_threshold = iou_threshold
         self.trackers = []
+        self.per_class = True
         self.frame_count = 0
         self.det_thresh = det_thresh
         self.delta_t = delta_t
@@ -371,6 +373,7 @@ def camera_update(self, trackers, warp_matrix):
         for tracker in trackers:
             tracker.camera_update(warp_matrix)
 
+    @PerClassDecorator
     def update(self, dets, im):
         """
         Params: