更新py文件

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

+import cv2
+import mediapipe as mp
+import time
+if __name__=='__main__':
+    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()
+
+
+