全身检测（视频）.py

import cv2
import mediapipe as mp

# https://blog.csdn.net/luozhichengaichenlei/article/details/117262688
# mp.solutions.drawing_utils用于绘制
mp_drawing = mp.solutions.drawing_utils

#参数：1、颜色，2、线条粗细，3、点的半径
DrawingSpec_point = mp_drawing.DrawingSpec((0, 255, 0), 1 , 1)
DrawingSpec_line = mp_drawing.DrawingSpec((0, 0, 255), 1, 1)

#mp.solutions.holistic是一个类别，是人的整体
mp_holistic = mp.solutions.holistic

#参数：1、是否检测静态图片，2、姿态模型的复杂度，3、结果看起来平滑（用于video有效），4、检测阈值，5、跟踪阈值
holistic = mp_holistic.Holistic(min_detection_confidence=0.5, min_tracking_confidence=0.5)
input_file=0
# input_file='input.mp4'

# INFO: Created TensorFlow Lite XNNPACK delegate for CPU.
#     Traceback (most recent call last):
# File "G:/project/pythonProj/pose/Pose_estimation-main/全身检测（视频）.py", line 38, in <module>
# image, results.face_landmarks, mp_holistic.FACE_CONNECTIONS, DrawingSpec_point, DrawingSpec_line)
# AttributeError: module 'mediapipe.python.solutions.holistic' has no attribute 'FACE_CONNECTIONS'
# [ WARN:1] global C:\Users\runneradmin\AppData\Local\Temp\pip-req-build-_xlv4eex\opencv\modules\videoio\src\cap_msmf.cpp (438) `anonymous-namespace'::SourceReaderCB::~SourceReaderCB terminating async callback
# help(mp)
# print(help(mp))
# mp_version=mp.__version__
# 没有
# print(mp_version)
cap = cv2.VideoCapture(input_file)
while cap.isOpened():
    success, image = cap.read()
    if not success:
        print("Ignoring empty camera frame.")
        continue
    image1 = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

    # 处理RGB图像
    results = holistic.process(image1)

    '''
    mp_holistic.PoseLandmark类中共33个人体骨骼点
    mp_holistic.HandLandmark类中共21个手部关键点
    脸部有468个关键点
    '''

    # 绘制
    try:
        connections=mp_holistic.FACE_CONNECTIONS
    except AttributeError as e:
        # print(e.__cause__)
        connections=mp_holistic.FACEMESH_TESSELATION
    # if connections is None:
    #     connections=
    mp_drawing.draw_landmarks(
        image, results.face_landmarks, connections, DrawingSpec_point, DrawingSpec_line)
    # mp_drawing.draw_landmarks(
    #     image, results.face_landmarks, mp_holistic.FACE_CONNECTIONS, DrawingSpec_point, DrawingSpec_line)
    # FACEMESH_TESSELATION
    mp_drawing.draw_landmarks(
        image, results.left_hand_landmarks, mp_holistic.HAND_CONNECTIONS, DrawingSpec_point, DrawingSpec_line)
    mp_drawing.draw_landmarks(
        image, results.right_hand_landmarks, mp_holistic.HAND_CONNECTIONS, DrawingSpec_point, DrawingSpec_line)
    mp_drawing.draw_landmarks(
        image, results.pose_landmarks, mp_holistic.POSE_CONNECTIONS, DrawingSpec_point, DrawingSpec_line)

    cv2.imshow('MediaPipe Holistic', image)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

holistic.close()
cv2.destroyAllWindows()
cap.release()