From 6b54f2398ca3467d99cbd75ad83820cd9d4c741c Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Fri, 5 Jul 2024 23:03:01 +0200 Subject: [PATCH] - fixed min/max calcualtion - improved readability of screen text --- ocv_corner_tracker.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/ocv_corner_tracker.py b/ocv_corner_tracker.py index 4d6a606..0472bc2 100644 --- a/ocv_corner_tracker.py +++ b/ocv_corner_tracker.py @@ -343,8 +343,8 @@ if __name__ == '__main__': print('Cannot read video file') key_wait = -1 - dist_min = [[0, 0]]*len(tracker_list) - dist_max = [[0, 0]]*len(tracker_list) + dist_min = [(0, 0)]*len(tracker_list) + dist_max = [(0, 0)]*len(tracker_list) while True: # Start timer timer = cv2.getTickCount() @@ -353,16 +353,15 @@ if __name__ == '__main__': if ok: image_anno = image.copy() i = 0 + cv2.rectangle(image_anno, (25, 0), (int(image_anno.shape[1]), 25*(1+len(tracker_list))), (0, 0, 0), -1) for ct in tracker_list: mean_distance = ct.process(image, image_anno) if mean_distance is not None: scaled_distance = (scale*mean_distance[0], scale*mean_distance[1]) - dist_min[i][0] = min(scaled_distance[0], dist_min[i][0]) - dist_min[i][1] = min(scaled_distance[1], dist_min[i][1]) - dist_max[i][0] = max(scaled_distance[0], dist_max[i][0]) - dist_max[i][1] = max(scaled_distance[1], dist_max[i][1]) + dist_min[i] = (min(scaled_distance[0], dist_min[i][0]), min(scaled_distance[1], dist_min[i][1])) + dist_max[i] = (max(scaled_distance[0], dist_max[i][0]), max(scaled_distance[1], dist_max[i][1])) cv2.putText(image_anno, f"Distance [{i}] : ({scaled_distance[0]:+05.2f}, {scaled_distance[1]:+05.2f}), Min: ({dist_min[i][0]:+05.2f}, {dist_min[i][1]:+05.2f}), Max: ({dist_max[i][0]:+05.2f}, {dist_max[i][1]:+05.2f})", - (20, 20 + 20 * i), cv2.FONT_HERSHEY_SIMPLEX, 0.8, ct.color, 2) + (25, 25 * (i + 1)), cv2.FONT_HERSHEY_SIMPLEX, 0.5, ct.color, 1) i += 1 # Calculate Frames per second (FPS)