fix(pose): 修正正面照被误判为1003(solvePnP翻转解)
ITERATIVE 偶发收敛到相机后方(tz<0),roll≈±180° 超阈值, 把正面照误判为非正面。检测到负深度时回退 SQPNP 重解正深度解。 补充回归测试。 Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
+10
-1
@@ -50,12 +50,21 @@ def estimate_head_pose(landmarks, image_width, image_height):
|
||||
[0, 0, 1]], dtype=np.float64)
|
||||
dist = np.zeros((4, 1)) # 假设无畸变
|
||||
|
||||
success, rvec, _tvec = cv2.solvePnP(
|
||||
success, rvec, tvec = cv2.solvePnP(
|
||||
_MODEL_POINTS, image_points, cam_matrix, dist,
|
||||
flags=cv2.SOLVEPNP_ITERATIVE,
|
||||
)
|
||||
if not success:
|
||||
return None
|
||||
# ITERATIVE 偶发收敛到相机后方的翻转解(tz<0),此时 roll 落在 ±180° 附近,
|
||||
# 会把真正的正面照误判为 1003。改用 SQPNP 重解正深度解。
|
||||
if float(tvec[2, 0]) < 0:
|
||||
ok2, rvec2, tvec2 = cv2.solvePnP(
|
||||
_MODEL_POINTS, image_points, cam_matrix, dist,
|
||||
flags=cv2.SOLVEPNP_SQPNP,
|
||||
)
|
||||
if ok2 and float(tvec2[2, 0]) > 0:
|
||||
rvec = rvec2
|
||||
rot, _ = cv2.Rodrigues(rvec)
|
||||
# 在「相机坐标系」(x右 y下 z内) 下抽取 Tait-Bryan 欧拉角,物理含义对齐:
|
||||
# yaw = 绕 Y(竖轴)转 → 左右扭头
|
||||
|
||||
Reference in New Issue
Block a user