Add ConfTrack Object Tracker

supervision/__init__.py

@@ -71,6 +71,7 @@
 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.tracker.confidence_tracker.core import ConfTrack
 from supervision.utils.file import list_files_with_extensions
 from supervision.utils.image import ImageSink, crop_image, place_image, resize_image
 from supervision.utils.notebook import plot_image, plot_images_grid

supervision/tracker/confidence_tracker/basetrack.py

@@ -0,0 +1,61 @@
+from collections import OrderedDict
+
+import numpy as np
+
+
+class TrackState(object):
+ New = 0
+ Tracked = 1
+ Lost = 2
+ LongLost = 3
+ Removed = 4
+
+
+class BaseTrack(object):
+ _count = 0
+
+ track_id = 0
+ is_activated = False
+ state = TrackState.New
+
+ history = OrderedDict()
+ features = []
+ curr_feature = None
+ score = 0
+ start_frame = 0
+ frame_id = 0
+ time_since_update = 0
+
+ # multi-camera
+ location = (np.inf, np.inf)
+
+ @property
+ def end_frame(self):
+ return self.frame_id
+
+ @staticmethod
+ def next_id():
+ 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_long_lost(self):
+ self.state = TrackState.LongLost
+
+ def mark_removed(self):
+ self.state = TrackState.Removed
+
+ @staticmethod
+ def clear_count():
+ BaseTrack._count = 0