Spaces:
Sleeping
Sleeping
File size: 1,321 Bytes
02f8487 |
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 38 39 40 41 42 |
# -*- coding: utf-8 -*-
# @Time : 2024/8/7 9:00
# @Author : shaoguowen
# @Email : [email protected]
# @Project : FasterLivePortrait
# @FileName: mediapipe_face_model.py
import cv2
import mediapipe as mp
import numpy as np
class MediaPipeFaceModel:
"""
MediaPipeFaceModel
"""
def __init__(self, **kwargs):
mp_face_mesh = mp.solutions.face_mesh
self.face_mesh = mp_face_mesh.FaceMesh(
static_image_mode=True,
max_num_faces=1,
refine_landmarks=True,
min_detection_confidence=0.5)
def predict(self, *data):
img_bgr = data[0]
img_rgb = cv2.cvtColor(img_bgr, cv2.COLOR_BGR2RGB)
h, w = img_bgr.shape[:2]
results = self.face_mesh.process(cv2.cvtColor(img_rgb, cv2.COLOR_BGR2RGB))
# Print and draw face mesh landmarks on the image.
if not results.multi_face_landmarks:
return []
outs = []
for face_landmarks in results.multi_face_landmarks:
landmarks = []
for landmark in face_landmarks.landmark:
# 提取每个关键点的 x, y, z 坐标
landmarks.append([landmark.x * w, landmark.y * h])
outs.append(np.array(landmarks))
return outs
|