fix per class

mikel-brostrom · Aug 22, 2023 · b590ce3 · b590ce3
1 parent d7527db
commit b590ce3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/boxmot/trackers/hybridsort/hybridsort.py b/boxmot/trackers/hybridsort/hybridsort.py
@@ -340,7 +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.per_class = True
         self.frame_count = 0
         self.det_thresh = det_thresh
         self.delta_t = delta_t

diff --git a/boxmot/utils/__init__.py b/boxmot/utils/__init__.py
@@ -30,7 +30,7 @@ def wrapper(*args, **kwargs):
             im = modified_args[1]
 
             # input one class of detections at a time in order to not mix them up
-            if instance.per_class[0] is True and dets.size != 0:
+            if instance.per_class is True and dets.size != 0:
                 dets_dict = {
                     class_id: np.array([det for det in dets if det[5] == class_id])
                     for class_id in set(det[5] for det in dets)