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

- print original and refined corner info

Browse Source 
- new key commands: ESC=quit, g=run, s=single step
This commit is contained in:
jens
2024-07-03 20:31:33 +02:00
parent 166f7e771e
commit 564719abcb
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ocv_template_matching.py
+10 -3
View File
@@ -14,11 +14,12 @@ def refine_corner(src_gray, corners):
criteria = (cv2.TERM_CRITERIA_EPS + cv2.TermCriteria_COUNT, 40, 0.001) criteria = (cv2.TERM_CRITERIA_EPS + cv2.TermCriteria_COUNT, 40, 0.001)
# Calculate the refined corner locations # 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 # Write them down
for i in range(corners.shape[0]): 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) COLOR_TRACKER = (0, 255, 0)
@@ -48,6 +49,8 @@ crop = image_crop(image.copy(), bb_crop)
ct_list = [] ct_list = []
count = 1 count = 1
key_wait = -1
while True: while True:
print(f"Add Corner {count}") print(f"Add Corner {count}")
bb_template = cv2.selectROI("Image", crop, fromCenter=True, showCrosshair=True) bb_template = cv2.selectROI("Image", crop, fromCenter=True, showCrosshair=True)
@@ -176,6 +179,10 @@ while True:
cv2.imshow("Image", image_anno) cv2.imshow("Image", image_anno)
# Exit if ESC pressed # Exit if ESC pressed
k = cv2.waitKey(-1) & 0xff k = cv2.waitKey(key_wait) & 0xff
if k == 27: if k == 27:
break break
if k == ord('g'):
key_wait = 30
if k == ord('s'):
key_wait = -1