pipeline/trackers/bytetrack.py

import supervision as sv


class ByteTrack:
    def __init__(self):
        self.tracker = sv.ByteTrack()
        self.smoother = sv.DetectionsSmoother()

    def __call__(self, detections):
        """Process detections using ByteTrack."""
        # supervision_detections = self._convert_pytorch_to_supervision(detections)
        tracked_detections = self.tracker.update_with_detections(detections)
        smoothed_detections = self.smoother.update_with_detections(tracked_detections)
        return smoothed_detections