利用OpenPose实现视频中人体骨骼角度的计算

import cv2
import numpy as np
from openpose import pyopenpose as op

# 配置OpenPose参数
params = dict()
params["model_folder"] = "path/to/openpose/models/"

# 启动OpenPose
opWrapper = op.WrapperPython()
opWrapper.configure(params)
opWrapper.start()

# 打开视频文件
cap = cv2.VideoCapture('input_video.mp4')

while cap.isOpened():
    ret, frame = cap.read()
    if not ret:
        break

    # 处理帧
    datum = op.Datum()
    datum.cvInputData = frame
    opWrapper.emplaceAndPop([datum])

    # 获取关键点
    keypoints = datum.poseKeypoints

    if keypoints is not None:
        for person in keypoints:
            for i in range(len(person)):
                if i+2 < len(person):
                    # 计算角度（例如，计算肘关节的角度）
                    x1, y1 = person[i]
                    x2, y2 = person[i+1]
                    x3, y3 = person[i+2]
                    angle = np.arctan2(y3-y2, x3-x2) - np.arctan2(y1-y2, x1-x2)
                    angle = np.degrees(angle)
                    print(f"Angle at keypoint {i}: {angle}")

    # 显示结果
    cv2.imshow('OpenPose Output', datum.cvOutputData)

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()