Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 \
|