基于mediapipe的单人人体骨架细节提取

最新推荐文章于 2024-09-05 23:24:54 发布

2301_79655511

最新推荐文章于 2024-09-05 23:24:54 发布

阅读量994

点赞数 18

文章标签： spring maven struts kafka spring boot java-ee intellij-idea

本文链接：https://blog.csdn.net/2301_79655511/article/details/136599109

版权

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,