jens/ocv_template_matching
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

- fixed min/max calcualtion

Browse Source 
- improved readability of screen text
This commit is contained in:
jens
2024-07-05 23:03:01 +02:00
parent eb8810b82a
commit 6b54f2398c
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ocv_corner_tracker.py
+6 -7
View File
@@ -343,8 +343,8 @@ if __name__ == '__main__':
print('Cannot read video file') print('Cannot read video file')
key_wait = -1 key_wait = -1
dist_min = [[0, 0]]*len(tracker_list) dist_min = [(0, 0)]*len(tracker_list)
dist_max = [[0, 0]]*len(tracker_list) dist_max = [(0, 0)]*len(tracker_list)
while True: while True:
# Start timer # Start timer
timer = cv2.getTickCount() timer = cv2.getTickCount()
@@ -353,16 +353,15 @@ if __name__ == '__main__':
if ok: if ok:
image_anno = image.copy() image_anno = image.copy()
i = 0 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: for ct in tracker_list:
mean_distance = ct.process(image, image_anno) mean_distance = ct.process(image, image_anno)
if mean_distance is not None: if mean_distance is not None:
scaled_distance = (scale*mean_distance[0], scale*mean_distance[1]) 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] = (min(scaled_distance[0], dist_min[i][0]), min(scaled_distance[1], dist_min[i][1]))
dist_min[i][1] = 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]))
dist_max[i][0] = max(scaled_distance[0], dist_max[i][0])
dist_max[i][1] = 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})", 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 i += 1
# Calculate Frames per second (FPS) # Calculate Frames per second (FPS)