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