Tracker latest

docs/tracker/core.md

@@ -0,0 +1,4 @@
+## ByteTrack
+
+:::supervision.tracker.byte_tracker.core.ByteTrack
+

mkdocs.yml

@@ -37,6 +37,8 @@ nav:
         - Utils: detection/utils.md
         - Tools:
             - Polygon Zone: detection/tools/polygon_zone.md
+    - Trackers:
+        - Core: tracker/core.md
     - Dataset:
         - Core: dataset/core.md
     - Metrics:

pyproject.toml

@@ -41,6 +41,7 @@ pyyaml = "^6.0"
 pillow = "^9.4.0"
 opencv-python = { version = "^4.8.0.74", optional = true }
 opencv-python-headless = "^4.8.0.74"
+scipy = "^1.9.0"
 
 
 [tool.poetry.extras]

supervision/__init__.py

@@ -31,6 +31,7 @@
 from supervision.geometry.core import Point, Position, Rect
 from supervision.geometry.utils import get_polygon_center
 from supervision.metrics.detection import ConfusionMatrix, MeanAveragePrecision
+from supervision.tracker.byte_tracker.core import ByteTrack
 from supervision.utils.file import list_files_with_extensions
 from supervision.utils.image import ImageSink, crop
 from supervision.utils.notebook import plot_image, plot_images_grid

supervision/tracker/byte_tracker/basetrack.py

@@ -0,0 +1,55 @@
+from collections import OrderedDict
+from enum import Enum
+
+import numpy as np
+
+
+class TrackState(Enum):
+    New = 0
+    Tracked = 1
+    Lost = 2
+    Removed = 3
+
+
+class BaseTrack:
+    _count = 0
+
+    def __init__(self):
+        self.track_id = 0
+        self.is_activated = False
+        self.state = TrackState.New
+
+        self.history = OrderedDict()
+        self.features = []
+        self.curr_feature = None
+        self.score = 0
+        self.start_frame = 0
+        self.frame_id = 0
+        self.time_since_update = 0
+
+        # multi-camera
+        self.location = (np.inf, np.inf)
+
+    @property
+    def end_frame(self) -> int:
+        return self.frame_id
+
+    @staticmethod
+    def next_id() -> int:
+        BaseTrack._count += 1
+        return BaseTrack._count
+
+    def activate(self, *args):
+        raise NotImplementedError
+
+    def predict(self):
+        raise NotImplementedError
+
+    def update(self, *args, **kwargs):
+        raise NotImplementedError
+
+    def mark_lost(self):
+        self.state = TrackState.Lost
+
+    def mark_removed(self):
+        self.state = TrackState.Removed