姿态估计py文件

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

 import cv2
 import mediapipe as mp
 import time
-if __name__=='__main__':
+if __name__ == '__main__':
     mpPose = mp.solutions.pose
     pose = mpPose.Pose()
     mpDraw = mp.solutions.drawing_utils
@@ -17,7 +17,8 @@ if __name__=='__main__':
             continue
         print(results.pose_landmarks)
         if results.pose_landmarks:
-            mpDraw.draw_landmarks(img, results.pose_landmarks, mpPose.POSE_CONNECTIONS)
+            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)
@@ -26,12 +27,10 @@ if __name__=='__main__':
         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.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()
-
-
-