From 564719abcb9adce0f42a1ce4319905ffc6808f50 Mon Sep 17 00:00:00 2001
From: Jens Ahrensfeld <jens@jayfield.org>
Date: Wed, 3 Jul 2024 20:31:33 +0200
Subject: [PATCH] - print original and refined corner info - new key commands:
 ESC=quit, g=run, s=single step

---
 ocv_template_matching.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/ocv_template_matching.py b/ocv_template_matching.py
index eeeb405..569f000 100644
--- a/ocv_template_matching.py
+++ b/ocv_template_matching.py
@@ -14,11 +14,12 @@ def refine_corner(src_gray, corners):
 	criteria = (cv2.TERM_CRITERIA_EPS + cv2.TermCriteria_COUNT, 40, 0.001)
 
 	# Calculate the refined corner locations
-	corners = cv2.cornerSubPix(src_gray, corners, winSize, zeroZone, criteria)
+	corners_refined = cv2.cornerSubPix(src_gray, corners.copy(), winSize, zeroZone, criteria)
 
 	# Write them down
 	for i in range(corners.shape[0]):
-		print(" -- Refined Corner [", i, "] (", corners[i, 0, 0], ",", corners[i, 0, 1], ")")
+		print(" -- Original Corner [", i, "] (", corners[i, 0, 0], ",", corners[i, 0, 1], ")")
+		print(" -- Refined Corner  [", i, "] (", corners_refined[i, 0, 0], ",", corners_refined[i, 0, 1], ")")
 
 
 COLOR_TRACKER = (0, 255, 0)
@@ -48,6 +49,8 @@ crop = image_crop(image.copy(), bb_crop)
 
 ct_list = []
 count = 1
+key_wait = -1
+
 while True:
 	print(f"Add Corner {count}")
 	bb_template = cv2.selectROI("Image", crop, fromCenter=True, showCrosshair=True)
@@ -176,6 +179,10 @@ while True:
 		cv2.imshow("Image", image_anno)
 
 	# Exit if ESC pressed
-	k = cv2.waitKey(-1) & 0xff
+	k = cv2.waitKey(key_wait) & 0xff
 	if k == 27:
 		break
+	if k == ord('g'):
+		key_wait = 30
+	if k == ord('s'):
+		key_wait = -1