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-12-03 19:34:43 +01:00
parent facc115e30
commit b300ddb166
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ar_tag_pose/detector_board_main.py
+1 -1
View File
@@ -154,7 +154,7 @@ class Detector(abc.ABC):
for det in self.detector: for det in self.detector:
success, r_vec, t_vec = det.process(img) success, r_vec, t_vec = det.process(img)
if success: if success:
rot_e = to_euler(to_tf(r_vec, t_vec)) rot_e = to_euler(to_tf(r_vec, t_vec), axis_order="XYZ")
rot = [round(180.0 / math.pi * v, 1) for v in rot_e] rot = [round(180.0 / math.pi * v, 1) for v in rot_e]
cv.putText(img, f"{det}: Tilt: {rot[0]:.1f}, Roll: {rot[1]:.1f}, Azimuth: {rot[2]:.1f}", cv.putText(img, f"{det}: Tilt: {rot[0]:.1f}, Roll: {rot[1]:.1f}, Azimuth: {rot[2]:.1f}",
(30, y_pos), cv.FONT_HERSHEY_SIMPLEX, 0.75, (0, 255, 0), 2) (30, y_pos), cv.FONT_HERSHEY_SIMPLEX, 0.75, (0, 255, 0), 2)
ar_tag_pose/utils.py
+2 -2
View File
@@ -25,7 +25,7 @@ def to_tf(r_vec: np.ndarray, t_vec: np.ndarray) -> np.ndarray:
r_mat = np.vstack((r_mat, np.array([0, 0, 0, 1]))) r_mat = np.vstack((r_mat, np.array([0, 0, 0, 1])))
return r_mat return r_mat
def to_euler(t1: np.ndarray, t2: np.ndarray | None = None, euler_axis: str= "XYZ") -> np.ndarray: def to_euler(t1: np.ndarray, t2: np.ndarray | None = None, axis_order: str= "XYZ") -> np.ndarray:
# Transform t2 to t1 coordinate system # Transform t2 to t1 coordinate system
if t2 is None: if t2 is None:
tf = np.linalg.inv(t1) tf = np.linalg.inv(t1)
@@ -38,7 +38,7 @@ def to_euler(t1: np.ndarray, t2: np.ndarray | None = None, euler_axis: str= "XYZ
# Convert to euler angles # Convert to euler angles
r = R.from_rotvec(r_vec) r = R.from_rotvec(r_vec)
res = R.as_euler(r, euler_axis) res = R.as_euler(r, axis_order)
return res return res