From bd30a1007035a652de990f777f2c60910d6b2c09 Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Wed, 3 Jul 2024 20:42:58 +0200 Subject: [PATCH] Refactored --- ocv_template_matching.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/ocv_template_matching.py b/ocv_template_matching.py index 569f000..155d59b 100644 --- a/ocv_template_matching.py +++ b/ocv_template_matching.py @@ -7,18 +7,18 @@ from util import bbox_extend, bbox_round, image_crop, to_rect, bbox_add_position from ocv_corner_tracker import CornerTracker -def refine_corner(src_gray, corners): +def refine_corner(src_gray, corners_original: np.array): # Set the needed parameters to find the refined corners - winSize = (5, 5) - zeroZone = (-1, -1) + win_size = (5, 5) + zero_zone = (-1, -1) criteria = (cv2.TERM_CRITERIA_EPS + cv2.TermCriteria_COUNT, 40, 0.001) # Calculate the refined corner locations - corners_refined = cv2.cornerSubPix(src_gray, corners.copy(), winSize, zeroZone, criteria) + corners_refined = cv2.cornerSubPix(src_gray, corners_original.copy(), win_size, zero_zone, criteria) # Write them down - for i in range(corners.shape[0]): - print(" -- Original Corner [", i, "] (", corners[i, 0, 0], ",", corners[i, 0, 1], ")") + for i in range(corners_original.shape[0]): + print(" -- Original Corner [", i, "] (", corners_original[i, 0, 0], ",", corners_original[i, 0, 1], ")") print(" -- Refined Corner [", i, "] (", corners_refined[i, 0, 0], ",", corners_refined[i, 0, 1], ")")