fulyvv/pose2sim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

only display the outer corners in findCorners

Browse Source
This commit is contained in:
davidpagnon 2023-11-07 13:35:43 +01:00
parent 475f7df0fa
commit c53553d8a3
2 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
Pose2Sim/calibration.py
View File

@ -748,11 +748,9 @@ def findCorners(img_path, corner_nb, objp=[], show=True):
criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 30, 0.001) # stop refining after 30 iterations or if error less than 0.001px criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 30, 0.001) # stop refining after 30 iterations or if error less than 0.001px
img = cv2.imread(img_path) img = cv2.imread(img_path)
print('eh', img is None)
if img is None: if img is None:
with suppress_stdout_stderr(): cap = cv2.VideoCapture(img_path)
cap = cv2.VideoCapture(img_path) ret, img = cap.read()
ret, img = cap.read()
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
@ -768,9 +766,10 @@ def findCorners(img_path, corner_nb, objp=[], show=True):
cv2.drawChessboardCorners(img, corner_nb, imgp, ret) cv2.drawChessboardCorners(img, corner_nb, imgp, ret)
# Add corner index # Add corner index
for i, corner in enumerate(imgp): for i, corner in enumerate(imgp):
x, y = corner.ravel() if i in [0, corner_nb[0]-1, corner_nb[0]*(corner_nb[1]-1), corner_nb[0]*corner_nb[1] -1]:
cv2.putText(img, str(i+1), (int(x)-5, int(y)-5), cv2.FONT_HERSHEY_SIMPLEX, .8, (255, 255, 255), 7) x, y = corner.ravel()
cv2.putText(img, str(i+1), (int(x)-5, int(y)-5), cv2.FONT_HERSHEY_SIMPLEX, .8, (0,0,0), 2) cv2.putText(img, str(i+1), (int(x)-5, int(y)-5), cv2.FONT_HERSHEY_SIMPLEX, .8, (255, 255, 255), 7)
cv2.putText(img, str(i+1), (int(x)-5, int(y)-5), cv2.FONT_HERSHEY_SIMPLEX, .8, (0,0,0), 2)
# Visualizer and key press event handler # Visualizer and key press event handler
for var_to_delete in ['imgp_confirmed', 'objp_confirmed']: for var_to_delete in ['imgp_confirmed', 'objp_confirmed']:

2
setup.cfg
View File

@ -1,6 +1,6 @@
[metadata] [metadata]
name = pose2sim name = pose2sim
version = 0.4.11 version = 0.4.12
author = David Pagnon author = David Pagnon
author_email = contact@david-pagnon.com author_email = contact@david-pagnon.com
description = Perform a markerless kinematic analysis from multiple calibrated views as a unified workflow from an OpenPose input to an OpenSim result. description = Perform a markerless kinematic analysis from multiple calibrated views as a unified workflow from an OpenPose input to an OpenSim result.