MediaPipe 是一款由 Google Research 开发并开源的多媒体机器学习模型应用框架。在谷歌,一系列重要产品,如 、Google Lens、ARCore、Google Home 以及 ,都已深度整合了 MediaPipe。本文将介绍的为基于mediapipe的人体骨架提取方案。
1、mediapipe的安装
安装指令如下:
pip install mediapipe
官网地址:MediaPipe | Google for Developers
如果需要除了人体骨架提取以外的mediapipe的功能,可以参照官网内的demo进行编写。
2、demo编写
参照官网给的demo进行简要的更改,如下是对视频进行骨架提取,可根据需求更改为摄像头摄像或者照片。
import cv2
import time
import mediapipe as mp
mp_drawing = mp.solutions.drawing_utils
mp_drawing_styles = mp.solutions.drawing_styles
mp_holistic = mp.solutions.holistic
cap = cv2.VideoCapture('1_demo2.mp4') # 替换为视频路径
fps_start_time = time.time()
fps = 0
with mp_holistic.Holistic(
min_detection_confidence=0.5,