在代码中已经添加了计算人体腿与右手的角度的功能,并在输出视频中显示了角度值。具体实现可以参考以下代码:

# 计算腿与右手的角度
angle = math.degrees(math.atan2(right_wrist.y - right_ankle.y, right_wrist.x - right_ankle.x) -
                     math.atan2(right_knee.y - right_ankle.y, right_knee.x - right_ankle.x))

# 在输出图片上显示角度值
cv2.putText(image, "Angle: {:.2f}".format(angle), (50, 50),
            cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2)

这段代码首先使用math.atan2函数计算右手腕与右脚踝、右膝盖与右脚踝之间的夹角,然后使用math.degrees函数将弧度转换为角度。最后,使用cv2.putText函数将角度值添加到输出图片中。在每一帧处理完毕后,将该帧写入输出视频文件

import cv2import mediapipe as mp# 初始化MediaPipe的人体姿势模型mp_drawing = mpsolutionsdrawing_utilsmp_pose = mpsolutionspose# 打开输入视频文件cap = cv2VideoCapture1mp4# 获取输入视频的帧率和分辨率fps = intcapgetcv2CAP_PROP_FPSwidth

原文地址: https://www.cveoy.top/t/topic/cJAJ 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录