From 7f7d28179e22d0ef0c4e7b601ce21be2dd72dc4c Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Thu, 4 Jul 2024 20:51:13 +0200 Subject: [PATCH] round for tracker rect --- ocv_corner_tracker.py | 2 +- ocv_track.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ocv_corner_tracker.py b/ocv_corner_tracker.py index 0837667..fe9d0ad 100644 --- a/ocv_corner_tracker.py +++ b/ocv_corner_tracker.py @@ -126,7 +126,7 @@ class CornerTracker: matcher_rect = to_rect(matcher_bbox) cv2.rectangle(_image_anno, matcher_rect[0], matcher_rect[1], COLOR_MATCHER, 1) - tracker_rect = to_rect(self.tracking_bb) + tracker_rect = to_rect(bbox_round(self.tracking_bb)) cv2.rectangle(_image_anno, tracker_rect[0], tracker_rect[1], self.color, 1) def init_reference_frame(self, _image: np.array): diff --git a/ocv_track.py b/ocv_track.py index 7528cb9..413e472 100644 --- a/ocv_track.py +++ b/ocv_track.py @@ -58,7 +58,7 @@ trackers = { 'BOOSTING': cv2.legacy.TrackerBoosting, 'MEDIANFLOW': cv2.legacy.TrackerMedianFlow } -tracker_type = 'KCF' +tracker_type = 'MOSSE' tracker = trackers[tracker_type].create() video = cv2.VideoCapture("./data/IMG_1730.MP4")