如何姿态估计
还是和前面那个由于使用的是Mediapipe,进行检测
代码部分
import cv2
import mediapipe as mpif __name__ == '__main__':mp_pose = mp.solutions.posepose = mp_pose.Pose(static_image_mode=True,model_complexity=1,smooth_landmarks=True,# enable_segmentation=True,min_detection_confidence=0.5,min_tracking_confidence=0.5)drawing = mp.solutions.drawing_utils'''mp_pose.Pose()其参数:1)static_image_mode:静态图像还是连续帧视频;2)model_complexity:人体姿态估计模型,0表示速度最快,精度最低(三者之中),1表示速度中间,精度中间(三者之中),2表示速度最慢,精度最高(三者之中);3)smooth_landmarks:是否平滑关键点;4)enable_segmentation:是否对人体进行抠图;5)min_detection_confidence:检测置信度阈值;6)min_tracking_confidence:各帧之间跟踪置信度阈值;'''# read img BGR to RGBimg = cv2.imread("img.png")img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)cv2.imshow("input", img)results = pose.process(img)drawing.draw_landmarks(img, results.pose_landmarks, mp_pose.POSE_CONNECTIONS)cv2.imshow("keypoint", img)drawing.plot_landmarks(results.pose_world_landmarks, mp_pose.POSE_CONNECTIONS)cv2.waitKey(0)cv2.destroyAllWindows()