Piarasingh85 commited on
Commit
3e95a4f
·
verified ·
1 Parent(s): e6ded88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -0
app.py CHANGED
@@ -91,6 +91,19 @@ def classifyPose(landmarks, output_image, display=False):
91
  right_knee_angle = calculateAngle(landmarks[mp_pose.PoseLandmark.RIGHT_HIP.value],
92
  landmarks[mp_pose.PoseLandmark.RIGHT_KNEE.value],
93
  landmarks[mp_pose.PoseLandmark.RIGHT_ANKLE.value])
 
 
 
 
 
 
 
 
 
 
 
 
 
94
 
95
  # Check for Five-Pointed Star Pose
96
  if abs(landmarks[mp_pose.PoseLandmark.LEFT_WRIST.value].y - landmarks[mp_pose.PoseLandmark.LEFT_HIP.value].y) < 100 and \
 
91
  right_knee_angle = calculateAngle(landmarks[mp_pose.PoseLandmark.RIGHT_HIP.value],
92
  landmarks[mp_pose.PoseLandmark.RIGHT_KNEE.value],
93
  landmarks[mp_pose.PoseLandmark.RIGHT_ANKLE.value])
94
+ # Get coordinates
95
+ left_hip = [landmarks[mp_pose.PoseLandmark.LEFT_HIP.value].x, landmarks[mp_pose.PoseLandmark.LEFT_HIP.value].y]
96
+ left_knee = [landmarks[mp_pose.PoseLandmark.LEFT_KNEE.value].x, landmarks[mp_pose.PoseLandmark.LEFT_KNEE.value].y]
97
+ left_ankle = [landmarks[mp_pose.PoseLandmark.LEFT_ANKLE.value].x, landmarks[mp_pose.PoseLandmark.LEFT_ANKLE.value].y]
98
+
99
+ right_hip = [landmarks[mp_pose.PoseLandmark.RIGHT_HIP.value].x, landmarks[mp_pose.PoseLandmark.RIGHT_HIP.value].y]
100
+ right_knee = [landmarks[mp_pose.PoseLandmark.RIGHT_KNEE.value].x, landmarks[mp_pose.PoseLandmark.RIGHT_KNEE.value].y]
101
+ right_ankle = [landmarks[mp_pose.PoseLandmark.RIGHT_ANKLE.value].x, landmarks[mp_pose.PoseLandmark.RIGHT_ANKLE.value].y]
102
+
103
+ # Calculate angles
104
+ left_leg_angle = calculate_angle(left_hip, left_knee, left_ankle)
105
+ right_leg_angle = calculate_angle(right_hip, right_knee, right_ankle)
106
+
107
 
108
  # Check for Five-Pointed Star Pose
109
  if abs(landmarks[mp_pose.PoseLandmark.LEFT_WRIST.value].y - landmarks[mp_pose.PoseLandmark.LEFT_HIP.value].y) < 100 and \