fix box_iou_batch import

Author: SkalskiP

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "trackers"
-version = "2.0.1"
+version = "2.0.2rc0"
 description = "A unified library for object tracking featuring clean room re-implementations of leading multi-object tracking algorithms"
 readme = "README.md"
 authors = [
@@ -37,7 +37,7 @@ dependencies = [
     "aiofiles>=24.1.0",
     "aiohttp>=3.11.16",
     "numpy>=2.0.2",
-    "supervision>=0.26.0rc7",
+    "supervision>=0.26.1",
     "tqdm>=4.67.1",
     "validators>=0.34.0",
     "scipy>=1.13.1",

trackers/utils/sort_utils.py

@@ -3,7 +3,6 @@
 
 import numpy as np
 import supervision as sv
-from supervision.detection.utils import box_iou_batch
 
 from trackers.core.deepsort.kalman_box_tracker import DeepSORTKalmanBoxTracker
 from trackers.core.sort.kalman_box_tracker import SORTKalmanBoxTracker
@@ -63,7 +62,7 @@ def get_iou_matrix(
         predicted_boxes = np.zeros((len(trackers), 4), dtype=np.float32)
 
     if len(trackers) > 0 and len(detection_boxes) > 0:
-        iou_matrix = box_iou_batch(predicted_boxes, detection_boxes)
+        iou_matrix = sv.box_iou_batch(predicted_boxes, detection_boxes)
     else:
         iou_matrix = np.zeros((len(trackers), len(detection_boxes)), dtype=np.float32)
 
@@ -109,7 +108,7 @@ def update_detections_with_track_ids(
         predicted_boxes = np.zeros((len(trackers), 4), dtype=np.float32)
 
     if len(trackers) > 0 and len(detection_boxes) > 0:
-        iou_matrix_final = box_iou_batch(predicted_boxes, detection_boxes)
+        iou_matrix_final = sv.box_iou_batch(predicted_boxes, detection_boxes)
 
     row_indices, col_indices = np.where(iou_matrix_final > minimum_iou_threshold)
     sorted_pairs = sorted(