jens/ocv_cam_pose
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactored

Browse Source
This commit is contained in:
jens
2025-11-24 12:30:41 +01:00
parent 8cfb6a44b7
commit 4ea570e2ab
2 changed files with 31 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cam_pose.py
+20 -16
View File
@@ -2,6 +2,9 @@ import numpy as np
import cv2 as cv
import glob
PAT_NX = 7
PAT_NY = 6
def draw_gizmo(img, corners, imgpts):
corner = tuple(corners[0].ravel().astype("int32"))
imgpts = imgpts.astype("int32")
@@ -26,6 +29,14 @@ def draw_cube(img, corners, imgpts):
return img
def init_data():
criteria = (cv.TERM_CRITERIA_EPS + cv.TERM_CRITERIA_MAX_ITER, 30, 0.001)
objp = np.zeros((PAT_NX * PAT_NY, 3), np.float32)
objp[:, :2] = np.mgrid[0:PAT_NX, 0:PAT_NY].T.reshape(-1, 2)
axis = np.float32([[3, 0, 0], [0, 3, 0], [0, 0, -3]]).reshape(-1, 3)
return criteria, objp, axis
def main():
process_video()
@@ -34,11 +45,8 @@ def process_video():
with np.load('results/cam.npz') as X:
mtx, dist, _, _ = [X[i] for i in ('mtx', 'dist', 'rvecs', 'tvecs')]
criteria = (cv.TERM_CRITERIA_EPS + cv.TERM_CRITERIA_MAX_ITER, 30, 0.001)
objp = np.zeros((6 * 7, 3), np.float32)
objp[:, :2] = np.mgrid[0:7, 0:6].T.reshape(-1, 2)
axis = np.float32([[3, 0, 0], [0, 3, 0], [0, 0, -3]]).reshape(-1, 3)
# Init
criteria, objp, axis = init_data()
# Open camera
video = cv.VideoCapture(0)
@@ -51,7 +59,7 @@ def process_video():
ok, img = video.read()
if not ok:
return
process_img(img, criteria, objp, axis, mtx, dist)
process(img, mtx, dist, criteria, objp, axis)
k = cv.waitKey(1) & 0xFF
if k == ord('q'):
break
@@ -60,25 +68,21 @@ def process_still():
with np.load('results/cam.npz') as X:
mtx, dist, _, _ = [X[i] for i in ('mtx', 'dist', 'rvecs', 'tvecs')]
criteria = (cv.TERM_CRITERIA_EPS + cv.TERM_CRITERIA_MAX_ITER, 30, 0.001)
objp = np.zeros((6 * 7, 3), np.float32)
objp[:, :2] = np.mgrid[0:7, 0:6].T.reshape(-1, 2)
axis = np.float32([[3, 0, 0], [0, 3, 0], [0, 0, -3]]).reshape(-1, 3)
# Init
criteria, objp, axis = init_data()
for fname in glob.glob('check*.jpg'):
img = cv.imread(fname)
process_img(img, criteria, objp, axis, mtx, dist)
process(img, mtx, dist, criteria, objp, axis)
k = cv.waitKey(1) & 0xFF
if k == ord('q'):
break
def process_img(img, criteria, objp, axis, mtx, dist): # Load previously saved data
pattern_size = (7, 6)
def process(img, mtx, dist, criteria, objp, axis): # Load previously saved data
gray = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
ret, corners = cv.findChessboardCorners(gray, pattern_size, corners=None, flags=cv.CALIB_CB_ADAPTIVE_THRESH+cv.CALIB_CB_NORMALIZE_IMAGE+cv.CALIB_CB_FAST_CHECK)
cv.drawChessboardCorners(img, corners=corners, patternSize=pattern_size, patternWasFound=ret)
ret, corners = cv.findChessboardCorners(gray, (PAT_NX, PAT_NY), corners=None, flags=cv.CALIB_CB_ADAPTIVE_THRESH+cv.CALIB_CB_NORMALIZE_IMAGE+cv.CALIB_CB_FAST_CHECK)
cv.drawChessboardCorners(img, corners=corners, patternSize=(PAT_NX, PAT_NY), patternWasFound=ret)
if ret:
corners2 = cv.cornerSubPix(gray, corners, (11, 11), (-1, -1), criteria)