- print original and refined corner info

- new key commands: ESC=quit, g=run, s=single step
This commit is contained in:
2024-07-03 20:31:33 +02:00
parent 166f7e771e
commit 564719abcb
+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)
# 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