| import numpy as np | |
| from numpy.ctypeslib import as_ctypes_type | |
| from face_util.faceutil import getPose68 | |
| import ctypes | |
| def get_face_pose(box, landmark): | |
| fw = box[2] - box[0] | |
| fh = box[3] - box[1] | |
| sz = pow(fw * fh, 0.5) | |
| cx = (box[2] + box[0]) / 2 | |
| cy = (box[3] + box[1]) / 2 | |
| x1 = cx - sz / 2 | |
| y1 = cy - sz / 2 | |
| rx = ctypes.c_float(0) | |
| ry = ctypes.c_float(0) | |
| rz = ctypes.c_float(0) | |
| landmark_vec = (ctypes.c_float * len(landmark))(*landmark) | |
| getPose68(x1, y1, sz, landmark_vec, 68, ctypes.pointer(rx), ctypes.pointer(ry), ctypes.pointer(rz)) | |
| return np.array([rx.value, ry.value, rz.value], dtype=np.float32) | |