- print original and refined corner info
- new key commands: ESC=quit, g=run, s=single step
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user