File size: 1,504 Bytes
e34aada |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
import numpy as np
import copy
yaw_idx_in_mediapipe_mesh = [356, 454, 361, 288, 397, 379, 378, 377, 152, 148, 149, 150, 172,58, 132, 234, 127]
brow_idx_in_mediapipe_mesh = [70, 63, 105, 66, 107, 336, 296, 334, 293, 300]
nose_idx_in_mediapipe_mesh = [6, 5, 1, 2, 129, 240, 2, 460, 358]
eye_idx_in_mediapipe_mesh = [33, 160, 158, 133, 153, 144, 362, 385, 387, 263, 373, 380]
mouth_idx_in_mediapipe_mesh = [78,191,80,81,82,13,312,311,310,415,308,324,318,402,317,14,87,178,88,95]
lm68_idx_in_mediapipe_mesh = yaw_idx_in_mediapipe_mesh + brow_idx_in_mediapipe_mesh + nose_idx_in_mediapipe_mesh + eye_idx_in_mediapipe_mesh + mouth_idx_in_mediapipe_mesh
def mediapipe_lm478_to_face_alignment_lm68(lm478, H, W, return_2d=True):
"""
lm478: [B, 478, 3] or [478,3]
"""
lm478 = copy.deepcopy(lm478)
lm478[..., 0] *= W
lm478[..., 1] *= H
n_dim = 2 if return_2d else False
if lm478.ndim == 2:
return lm478[lm68_idx_in_mediapipe_mesh, :n_dim].astype(np.int16)
elif lm478.ndim == 3:
return lm478[:, lm68_idx_in_mediapipe_mesh, :n_dim].astype(np.int16)
else:
raise ValueError("input lm478 ndim should in 2 or 3!")
def mediapipe_lm478_to_lm68_3d(lm478):
"""
lm478: [B, 478, 3] or [478,3]
also works for lm468
"""
if lm478.ndim == 2:
return lm478[lm68_idx_in_mediapipe_mesh]
elif lm478.ndim == 3:
return lm478[:, lm68_idx_in_mediapipe_mesh]
else:
raise ValueError("input lm478 ndim should in 2 or 3!") |