diff --git a/ocv_corner_tracker.py b/ocv_corner_tracker.py index 06ab4f4..7ab514f 100644 --- a/ocv_corner_tracker.py +++ b/ocv_corner_tracker.py @@ -135,6 +135,9 @@ class CornerTracker: tracker_rect = to_rect(bbox_round(self.tracking_bb)) cv2.rectangle(_image_anno, tracker_rect[0], tracker_rect[1], self.color, 1) + cv2.putText(_image_anno, f"{self.id}", (self.tracking_bb[0], self.tracking_bb[1] - 4), + cv2.FONT_HERSHEY_SIMPLEX, 0.8, self.color, 2) + @staticmethod def mask_init(_image: np.array): # Create mask