Update src/detection/strategies/geometric.py

src/detection/strategies/geometric.py

@@ -106,7 +106,7 @@ class GeometricProcessor(BaseProcessor):
                 face_3d = np.array([[0.0,0.0,0.0],[0.0,-330.0,-65.0],[-225.0,170.0,-135.0],[225.0,170.0,-135.0],[-150.0,-150.0,-125.0],[150.0,-150.0,-125.0]],dtype=np.float32)
                 face_2d = np.array([(landmarks[1].x*w,landmarks[1].y*h),(landmarks[152].x*w,landmarks[152].y*h),(landmarks[263].x*w,landmarks[263].y*h),(landmarks[33].x*w,landmarks[33].y*h),(landmarks[287].x*w,landmarks[287].y*h),(landmarks[57].x*w,landmarks[57].y*h)],dtype=np.float32)
                 cam_matrix = np.array([[w,0,w/2],[0,w,h/2],[0,0,1]],dtype=np.float32)
-                _, rvec, _ = cv2.solvePnP(face_3d, face_2d, cam_mtx, self.zeros_4x1, flags=cv2.SOLVEPNP_EPNP)
+                _, rvec, _ = cv2.solvePnP(face_3d, face_2d, cam_matrix, self.zeros_4x1, flags=cv2.SOLVEPNP_EPNP)
                 rmat, _ = cv2.Rodrigues(rvec)
                 angles, _, _, _, _, _ = cv2.RQDecomp3x3(rmat)
                 pitch, yaw = angles[0], angles[1]