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

Refactored

Browse Source
This commit is contained in:
jens
2024-07-03 20:42:58 +02:00
parent 564719abcb
commit bd30a10070
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ocv_template_matching.py
+6 -6
View File
@@ -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], ")")