Add OC-SORT tracker support and align dependency versions

Integrate `OCSORTTracker` with direction consistency weight and `delta_t`
controls.

Update tracker parameter descriptions to show which trackers use each
setting.

Bump:
- `trackers` to `2.3.0`
- `inference-models` to `>=0.19.0`

app.py

@@ -13,7 +13,7 @@ import torch
 from tqdm import tqdm
 from inference_models import AutoModel
 
-from trackers import ByteTrackTracker, SORTTracker, frames_from_source
+from trackers import ByteTrackTracker, OCSORTTracker, SORTTracker, frames_from_source
 
 MAX_DURATION_SECONDS = 30
 
@@ -28,7 +28,7 @@ MODELS = [
     "rfdetr-seg-large",
 ]
 
-TRACKERS = ["bytetrack", "sort"]
+TRACKERS = ["bytetrack", "sort", "ocsort"]
 
 COCO_CLASSES = [
     "person",
@@ -147,13 +147,15 @@ VIDEO_EXAMPLES = [
     [
         "https://storage.googleapis.com/com-roboflow-marketing/supervision/video-examples/bikes-1280x720-1.mp4",
         "rfdetr-small",
-        "bytetrack",
+        "ocsort",
         0.2,
         30,
         0.3,
         3,
         0.1,
         0.6,
+        0.2,
+        3,
         [],
         "",
         True,
@@ -166,13 +168,15 @@ VIDEO_EXAMPLES = [
     [
         "https://storage.googleapis.com/com-roboflow-marketing/supervision/video-examples/bikes-1280x720-1.mp4",
         "rfdetr-small",
-        "bytetrack",
+        "ocsort",
         0.2,
         30,
         0.3,
         3,
         0.1,
         0.6,
+        0.2,
+        3,
         ["person"],
         "",
         True,
@@ -192,6 +196,8 @@ VIDEO_EXAMPLES = [
         3,
         0.3,
         0.6,
+        0.2,
+        3,
         [],
         "",
         True,
@@ -211,6 +217,8 @@ VIDEO_EXAMPLES = [
         3,
         0.1,
         0.6,
+        0.2,
+        3,
         [],
         "",
         True,
@@ -230,6 +238,8 @@ VIDEO_EXAMPLES = [
         3,
         0.1,
         0.6,
+        0.2,
+        3,
         [],
         "",
         True,
@@ -249,6 +259,8 @@ VIDEO_EXAMPLES = [
         3,
         0.1,
         0.6,
+        0.2,
+        3,
         [],
         "",
         True,
@@ -268,6 +280,8 @@ VIDEO_EXAMPLES = [
         3,
         0.1,
         0.6,
+        0.2,
+        3,
         [],
         "1",
         True,
@@ -287,6 +301,8 @@ VIDEO_EXAMPLES = [
         3,
         0.1,
         0.6,
+        0.2,
+        3,
         [],
         "",
         True,
@@ -299,13 +315,15 @@ VIDEO_EXAMPLES = [
     [
         "https://storage.googleapis.com/com-roboflow-marketing/supervision/video-examples/vehicles-1280x720.mp4",
         "rfdetr-medium",
-        "bytetrack",
+        "ocsort",
         0.2,
         30,
         0.3,
         3,
         0.1,
         0.6,
+        0.2,
+        3,
         [],
         "",
         True,
@@ -383,6 +401,8 @@ def track(
     minimum_consecutive_frames: int,
     minimum_iou_threshold: float,
     high_conf_det_threshold: float,
+    direction_consistency_weight: float,
+    delta_t: int,
     classes: list[str] | None = None,
     track_ids: str = "",
     show_boxes: bool = True,
@@ -419,6 +439,15 @@ def track(
             minimum_iou_threshold=minimum_iou_threshold,
             high_conf_det_threshold=high_conf_det_threshold,
         )
+    elif tracker_type == "ocsort":
+        tracker = OCSORTTracker(
+            lost_track_buffer=lost_track_buffer,
+            minimum_consecutive_frames=minimum_consecutive_frames,
+            minimum_iou_threshold=minimum_iou_threshold,
+            high_conf_det_threshold=high_conf_det_threshold,
+            direction_consistency_weight=direction_consistency_weight,
+            delta_t=delta_t,
+        )
     else:
         tracker = SORTTracker(
             lost_track_buffer=lost_track_buffer,
@@ -558,7 +587,7 @@ with gr.Blocks(title="Trackers Playground 🔥") as demo:
                     value=30,
                     step=1,
                     label="Lost Track Buffer",
-                    info="Frames to keep a lost track before removing it.",
+                    info="Frames to keep a lost track before removing it (ByteTrack, SORT, OC-SORT).",
                 )
                 track_activation_slider = gr.Slider(
                     minimum=0.0,
@@ -566,7 +595,7 @@ with gr.Blocks(title="Trackers Playground 🔥") as demo:
                     value=0.3,
                     step=0.05,
                     label="Track Activation Threshold",
-                    info="Minimum score for a track to be activated.",
+                    info="Minimum score for a track to be activated (ByteTrack, SORT).",
                 )
                 minimum_consecutive_slider = gr.Slider(
                     minimum=1,
@@ -574,7 +603,7 @@ with gr.Blocks(title="Trackers Playground 🔥") as demo:
                     value=2,
                     step=1,
                     label="Minimum Consecutive Frames",
-                    info="Detections needed before a track is confirmed.",
+                    info="Detections needed before a track is confirmed (ByteTrack, SORT, OC-SORT).",
                 )
                 minimum_iou_slider = gr.Slider(
                     minimum=0.0,
@@ -582,7 +611,7 @@ with gr.Blocks(title="Trackers Playground 🔥") as demo:
                     value=0.1,
                     step=0.05,
                     label="Minimum IoU Threshold",
-                    info="Overlap required to match a detection to a track.",
+                    info="Overlap required to match a detection to a track (ByteTrack, SORT, OC-SORT).",
                 )
                 high_confidence_slider = gr.Slider(
                     minimum=0.0,
@@ -590,7 +619,23 @@ with gr.Blocks(title="Trackers Playground 🔥") as demo:
                     value=0.6,
                     step=0.05,
                     label="High Confidence Detection Threshold",
-                    info="Detections above this are matched first (ByteTrack only).",
+                    info="Detections above this are matched first (ByteTrack / OC-SORT).",
+                )
+                direction_consistency_slider = gr.Slider(
+                    minimum=0.0,
+                    maximum=1.0,
+                    value=0.2,
+                    step=0.05,
+                    label="Direction Consistency Weight",
+                    info="Weight for direction consistency in association cost (OC-SORT only).",
+                )
+                delta_t_slider = gr.Slider(
+                    minimum=1,
+                    maximum=10,
+                    value=3,
+                    step=1,
+                    label="Delta T",
+                    info="Past frames for velocity estimation during occlusion (OC-SORT only).",
                 )
 
             with gr.Column():
@@ -640,6 +685,8 @@ with gr.Blocks(title="Trackers Playground 🔥") as demo:
             minimum_consecutive_slider,
             minimum_iou_slider,
             high_confidence_slider,
+            direction_consistency_slider,
+            delta_t_slider,
             class_filter,
             track_id_filter,
             show_boxes_checkbox,
@@ -664,6 +711,8 @@ with gr.Blocks(title="Trackers Playground 🔥") as demo:
             minimum_consecutive_slider,
             minimum_iou_slider,
             high_confidence_slider,
+            direction_consistency_slider,
+            delta_t_slider,
             class_filter,
             track_id_filter,
             show_boxes_checkbox,

requirements.txt

@@ -1,3 +1,3 @@
 gradio>=6.3.0,<6.4.0
-inference-models==0.18.6rc14
-trackers==2.2.0rc1
+inference-models>=0.19.0
+trackers==2.3.0