Hugging Face's logo

Spaces:
amarkc
/
yoga-pose-detection-correction
Runtime error

App Files Community
yoga-pose-detection-correction
File size: 4,057 Bytes 
c8f49a6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
 
import numpy as np


def angle(p1, p2, p3):
    a = np.array([p1[0], p1[1]])
    b = np.array([p2[0], p2[1]])
    c = np.array([p3[0], p3[1]])

    radians = np.arctan2(c[1] - b[1], c[0] - b[0]) - \
        np.arctan2(a[1] - b[1], a[0] - b[0])
    angle = np.abs(radians * 180.0 / np.pi)

    if angle > 180:
        angle = 360 - angle

    return angle


def rangles(pose_df, landmarks_points):
    for index, row in pose_df.iterrows():
        landmarks_points["left_shoulder"] = [
            row["left_shoulder_x"], row["left_shoulder_y"]]

        landmarks_points["right_shoulder"] = [
            row["right_shoulder_x"], row["right_shoulder_y"]]

        landmarks_points["left_elbow"] = [
            row["left_elbow_x"], row["left_elbow_y"]]

        landmarks_points["right_elbow"] = [
            row["right_elbow_x"], row["right_elbow_y"]]

        landmarks_points["left_wrist"] = [
            row["left_wrist_x"], row["left_wrist_y"]]

        landmarks_points["right_wrist"] = [
            row["right_wrist_x"], row["right_wrist_y"]]

        landmarks_points["left_hip"] = [
            row["left_hip_x"], row["left_hip_y"]]

        landmarks_points["right_hip"] = [
            row["right_hip_x"], row["right_hip_y"]]

        landmarks_points["left_knee"] = [
            row["left_knee_x"], row["left_knee_y"]]

        landmarks_points["right_knee"] = [
            row["right_knee_x"], row["right_knee_y"]]

        landmarks_points["left_ankle"] = [
            row["left_ankle_x"], row["left_ankle_y"]]

        landmarks_points["right_ankle"] = [
            row["right_ankle_x"], row["right_ankle_y"]]

        landmarks_points["left_heel"] = [
            row["left_heel_x"], row["left_heel_y"]]

        landmarks_points["right_heel"] = [
            row["right_heel_x"], row["right_heel_y"]]

        landmarks_points["left_foot_index"] = [
            row["left_foot_index_x"], row["left_foot_index_y"]]

        landmarks_points["right_foot_index"] = [
            row["right_foot_index_x"], row["right_foot_index_y"]]

        armpit_left = angle(
            landmarks_points["left_elbow"],
            landmarks_points["left_shoulder"],
            landmarks_points["left_hip"]
        )
        armpit_right = angle(
            landmarks_points["right_elbow"],
            landmarks_points["right_shoulder"],
            landmarks_points["right_hip"]
        )

        elbow_left = angle(
            landmarks_points["left_shoulder"],
            landmarks_points["left_elbow"],
            landmarks_points["left_wrist"]
        )
        elbow_right = angle(
            landmarks_points["right_shoulder"],
            landmarks_points["right_elbow"],
            landmarks_points["right_wrist"]
        )

        hip_left = angle(
            landmarks_points["right_hip"],
            landmarks_points["left_hip"],
            landmarks_points["left_knee"]
        )
        hip_right = angle(
            landmarks_points["left_hip"],
            landmarks_points["right_hip"],
            landmarks_points["right_knee"]
        )

        knee_left = angle(
            landmarks_points["left_hip"],
            landmarks_points["left_knee"],
            landmarks_points["left_ankle"]
        )
        knee_right = angle(
            landmarks_points["right_hip"],
            landmarks_points["right_knee"],
            landmarks_points["right_ankle"]
        )

        ankle_left = angle(
            landmarks_points["left_knee"],
            landmarks_points["left_ankle"],
            landmarks_points["left_foot_index"]
        )
        ankle_right = angle(
            landmarks_points["right_knee"],
            landmarks_points["right_ankle"],
            landmarks_points["right_foot_index"]
        )

        angles = [
            armpit_left,
            armpit_right,
            elbow_left,
            elbow_right,
            hip_left,
            hip_right,
            knee_left,
            knee_right,
            ankle_left,
            ankle_right
        ]
        return angles