feat(trackers): Implement SORTTracker

[soumik12345]

Description

Create SORTTracker as an implementation of the SORT (Simple Online and Realtime Tracking).

Type of change

Please delete options that are not relevant.

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)
• This change requires a documentation update

Example Usage

import numpy as np
import supervision as sv
from rfdetr import RFDETRBase
from rfdetr.util.coco_classes import COCO_CLASSES
from trackers import SORTTracker


model = RFDETRBase(device="mps")
tracker = SORTTracker()

def callback(frame: np.ndarray, _: int):
    detections = model.predict(frame, threshold=0.5)
    detections = tracker.update_with_detections(detections)

    labels = [
        f"{COCO_CLASSES[class_id]} {confidence:.2f}"
        for class_id, confidence in zip(detections.class_id, detections.confidence)
    ]

    annotated_image = frame.copy()
    annotated_image = sv.BoxAnnotator().annotate(annotated_image, detections)
    annotated_image = sv.LabelAnnotator().annotate(annotated_image, detections, labels)

    return annotated_image

sv.process_video(
    source_path="data/traffic_video.mp4",
    target_path="data/out.mp4",
    callback=callback,
)

Docs

• Docs updated?