使用MediaPipe Python进行人脸识别

MediaPipe介绍

MediaPipe 是一款由 Google Research 开发并开源的多媒体机器学习模型应用框架。在谷歌，一系列重要产品，如 、Google Lens、ARCore、Google Home 以及 ，都已深度整合了 MediaPipe。

MediaPipe安装

我使用Ubuntu 20.04，Python 3.8 可以直接在命令行使用pip安装。

pip install opencv-contrib-python
pip install mediapipe

MediaPipe人脸识别Python代码

我使用的测试视频是618c1d1b9d4a407096f5a287_lookV5-transcode.mp4，也可以尝试webm格式的视频。

import cv2
import mediapipe as mp
import time


cap = cv2.VideoCapture('./618c1d1b9d4a407096f5a287_lookV5-transcode.mp4')
pTime = 0

mpDraw = mp.solutions.drawing_utils
mpFaceMesh = mp.solutions.face_mesh
faceMesh = mpFaceMesh.FaceMesh(max_num_faces=2)
drawSpec = mpDraw.DrawingSpec(thickness=1