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

3
Pose2Sim/calibration.py
View File

@ -748,9 +748,7 @@ 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)
@ -768,6 +766,7 @@ 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):
if i in [0, corner_nb[0]-1, corner_nb[0]*(corner_nb[1]-1), corner_nb[0]*corner_nb[1] -1]:
x, y = corner.ravel() x, y = corner.ravel()
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, (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) cv2.putText(img, str(i+1), (int(x)-5, int(y)-5), cv2.FONT_HERSHEY_SIMPLEX, .8, (0,0,0), 2)

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.