姿态估计py文件

data/1.OpenCV初阶/7.OpenCV中的深度学习/4.姿态估计/attitude_estimation.py

+import cv2
+import mediapipe as mp
+import time
+
+mpPose = mp.solutions.pose
+pose = mpPose.Pose()
+mpDraw = mp.solutions.drawing_utils
+cap = cv2.VideoCapture('1.mp4')
+pTime = 0
+while True:
+    success, img = cap.read()
+    if success is False:
+        break
+    imgRGB = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    results = pose.process(imgRGB)
+    if results is None:
+        continue
+    print(results.pose_landmarks)
+    if results.pose_landmarks:
+        mpDraw.draw_landmarks(img, results.pose_landmarks, mpPose.POSE_CONNECTIONS)
+    for id, lm in enumerate(results.pose_landmarks.landmark):
+        h, w, c = img.shape
+    print(id, lm)
+    cx, cy = int(lm.x * w), int(lm.y * h)
+    cv2.circle(img, (cx, cy), 5, (255, 0, 0), cv2.FILLED)
+    cTime = time.time()
+    fps = 1 / (cTime - pTime)
+    pTime = cTime
+    cv2.putText(img, str(int(fps)), (50, 50), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 0, 0), 3)
+    cv2.imshow("Image", img)
+    key = cv2.waitKey(1) & 0xFF
+# do a bit of cleanup
+cv2.destroyAllWindows()
+cap.release()
+