-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcalc_angles.py
136 lines (108 loc) · 3.96 KB
/
calc_angles.py
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
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