forked from shubham-goel/4D-Humans
-
Notifications
You must be signed in to change notification settings - Fork 1
/
joints.py
54 lines (53 loc) · 924 Bytes
/
joints.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
# dictionary needed for labeling joints
JOINT_NAMES = [
"Nose",
"Neck",
"RShoulder",
"RElbow",
"RWrist",
"LShoulder",
"LElbow",
"LWrist",
"MidHip",
"RHip",
"RKnee",
"RAnkle",
"LHip",
"LKnee",
"LAnkle",
"REye",
"LEye",
"REar",
"LEar",
"LBigToe",
"LSmallToe",
"LHeel",
"RBigToe",
"RSmallToe",
"RHeel",
"Right Ankle",
"Right Knee",
"Right Hip",
"Left Hip",
"Left Knee",
"Left Ankle",
"Right Wrist",
"Right Elbow",
"Right Shoulder",
"Left Shoulder",
"Left Elbow",
"Left Wrist",
"Neck (LSP)",
"Top of Head (LSP)",
"Pelvis (MPII)",
"Thorax (MPII)",
"Spine (H36M)",
"Jaw (H36M)",
"Head (H36M)",
"Nose",
"Left Eye",
"Right Eye",
"Left Ear",
"Right Ear",
]
joint_dict = {JOINT_NAMES[:45][i]: i for i in range(0, len(JOINT_NAMES[:45]))}