-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathzju_skeletons.py
148 lines (139 loc) · 6.29 KB
/
zju_skeletons.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
137
138
139
140
141
142
143
144
145
146
147
148
import numpy as np
coordinate_scale = 1.5 # WIM
bones = np.array([
[-1, 0], [0, 1], [0, 2], [0, 3], [1, 4], [2, 5],
[3, 6], [4, 7], [5, 8], [6, 9], [7, 10], [8, 11],
[9, 12], [9, 13], [9, 14], [12, 15], [13, 16], [14, 17],
[16, 18], [17, 19], [18, 20], [19, 21], [15, 22], [15, 23]])
joints_336 = np.array([
[ 0.13496284, 0.60437698, 0.91678499],
[ 0.07003798, 0.62519505, 0.82765674],
[ 0.14507963, 0.52823428, 0.82071441],
[ 0.12594378, 0.59946953, 1.02875119],
[ 0.13129449, 0.69651831, 0.46570324],
[ 0.09422848, 0.42355361, 0.48319366],
[ 0.1293368 , 0.6199149 , 1.15771843],
[ 0.10944361, 0.65188462, 0.07142071],
[ 0.04757433, 0.33130168, 0.09485329],
[ 0.16494826, 0.63418366, 1.20440684],
[ 0.2245293 , 0.70823528, 0.02113438],
[ 0.13817327, 0.24646245, 0.03944982],
[ 0.163633 , 0.62581851, 1.3717629 ],
[ 0.14051063, 0.67748064, 1.28706163],
[ 0.19122936, 0.59825736, 1.29102583],
[ 0.20419944, 0.65253123, 1.52597385],
[ 0.06350005, 0.77100887, 1.3400363 ],
[ 0.24772937, 0.51239244, 1.34312646],
[-0.15204673, 0.82688955, 1.1995117 ],
[ 0.34262291, 0.32630115, 1.17153185],
[-0.3199668 , 0.93027384, 1.04943668],
[ 0.47660445, 0.22664063, 0.99263883],
[ 0.196293 , 0.63788279, 1.52744017],
[ 0.26777623, 0.67148338, 1.57264974]], dtype=np.float32) / coordinate_scale
joints_377 = np.array([
[ 5.29272486e-02, 5.95825134e-01, 9.56801108e-01],
[ 5.43072272e-02, 6.56659359e-01, 8.63115328e-01],
[ 1.47068967e-02, 5.42014611e-01, 8.53627616e-01],
[ 2.17200996e-02, 6.06265630e-01, 1.06378111e+00],
[ 4.02930743e-02, 7.09995519e-01, 4.90189217e-01],
[-2.87120923e-02, 4.93337757e-01, 4.99575297e-01],
[ 1.38764322e-02, 6.20664357e-01, 1.19308711e+00],
[-2.04216075e-02, 7.03598285e-01, 9.48140267e-02],
[-8.92428792e-02, 4.96270322e-01, 9.94298026e-02],
[ 3.39795297e-02, 6.01200671e-01, 1.24595576e+00],
[ 1.02307729e-01, 7.35155422e-01, 4.03967070e-02],
[-8.91470969e-04, 4.08922453e-01, 4.35669674e-02],
[-7.57562263e-03, 6.06320080e-01, 1.40752151e+00],
[ 4.00391837e-02, 6.49048873e-01, 1.32902545e+00],
[ 2.59406155e-04, 5.64623472e-01, 1.32903002e+00],
[ 1.08151413e-02, 6.03902973e-01, 1.56849734e+00],
[ 6.60143213e-02, 7.65403062e-01, 1.38432433e+00],
[-5.08460713e-02, 4.76996866e-01, 1.38234794e+00],
[ 8.11337358e-02, 9.23543939e-01, 1.17316138e+00],
[-1.61370977e-01, 3.84886880e-01, 1.15231350e+00],
[ 3.10536472e-01, 8.70691851e-01, 1.25430852e+00],
[ 3.19768126e-03, 2.02920228e-01, 1.16867520e+00],
[ 1.02966497e-04, 5.91331896e-01, 1.56732889e+00],
[ 7.11017651e-02, 6.03947036e-01, 1.62245774e+00]], dtype=np.float32) / coordinate_scale
joints_381 = np.array([
[ 0.11624644, 0.53746076, 0.89089895],
[ 0.09419226, 0.60216242, 0.80318765],
[ 0.09048743, 0.4809103 , 0.78623713],
[ 0.08951046, 0.53084283, 0.99793969],
[ 0.17872877, 0.69349635, 0.45976228],
[ 0.12368623, 0.45784874, 0.43969546],
[ 0.085254 , 0.53386191, 1.12593402],
[-0.0050869 , 0.63857268, 0.12299054],
[ 0.05886385, 0.51064966, 0.05702631],
[ 0.11430919, 0.51802387, 1.17533156],
[ 0.06761884, 0.72232175, 0.04642154],
[ 0.17214291, 0.44823128, 0.02546152],
[ 0.08156318, 0.49973805, 1.33638394],
[ 0.11000371, 0.55959598, 1.26078093],
[ 0.09780648, 0.46778414, 1.25553864],
[ 0.1302978 , 0.50400031, 1.4882648 ],
[ 0.10241624, 0.67244858, 1.32366373],
[ 0.07853949, 0.36744271, 1.30297093],
[ 0.07274572, 0.8504242 , 1.13853918],
[ 0.06298235, 0.17208556, 1.12432953],
[ 0.13294445, 0.82237028, 1.37114309],
[ 0.17185479, 0.18184024, 1.33693648],
[ 0.125432 , 0.48832351, 1.48757621],
[ 0.19591499, 0.52640475, 1.52916139]], dtype=np.float32) / coordinate_scale
joints_384 = np.array([
[ 0.3120577 , 0.53106883, 0.9961997 ],
[ 0.28526594, 0.59311776, 0.9050824 ],
[ 0.27917979, 0.470626 , 0.89239332],
[ 0.29077566, 0.52905616, 1.10884281],
[ 0.27419927, 0.64862529, 0.52818945],
[ 0.27617157, 0.44805529, 0.52951726],
[ 0.29433114, 0.53755902, 1.24248986],
[ 0.22614057, 0.66800368, 0.12589747],
[ 0.24738027, 0.45955923, 0.12033803],
[ 0.32591961, 0.52291041, 1.29342221],
[ 0.34039857, 0.71245058, 0.05740424],
[ 0.35984467, 0.41002707, 0.05625168],
[ 0.30450217, 0.51087117, 1.46183195],
[ 0.32871422, 0.56910698, 1.37981441],
[ 0.31446494, 0.47440459, 1.37893007],
[ 0.32431368, 0.5165235 , 1.62452282],
[ 0.3273436 , 0.69061365, 1.43941111],
[ 0.29640024, 0.37101237, 1.4329838 ],
[ 0.30554443, 0.91106386, 1.28950803],
[ 0.23243131, 0.14019801, 1.29816735],
[ 0.42140607, 1.13383456, 1.25781745],
[ 0.22601591, -0.09614008, 1.21765824],
[ 0.31863594, 0.50053669, 1.62272184],
[ 0.37060074, 0.53831407, 1.68833364]], dtype=np.float32) / coordinate_scale
joints_387 = np.array([
[ 0.20169297, 0.4228984 , 0.94634726],
[ 0.19536159, 0.48698033, 0.85403625],
[ 0.16275624, 0.36903918, 0.84235091],
[ 0.17384651, 0.42821618, 1.05576618],
[ 0.20269658, 0.51979699, 0.47945587],
[ 0.13237837, 0.32561786, 0.48729902],
[ 0.17072693, 0.4386258 , 1.18685174],
[ 0.17103959, 0.49675883, 0.08159364],
[ 0.10970428, 0.32904192, 0.08362136],
[ 0.1949761 , 0.41861746, 1.2388339 ],
[ 0.29845172, 0.51727868, 0.03153984],
[ 0.18786733, 0.22936974, 0.03231457],
[ 0.1600011 , 0.41593911, 1.40340483],
[ 0.20182889, 0.46464939, 1.3237853 ],
[ 0.16727489, 0.37664774, 1.32271252],
[ 0.18480128, 0.40757661, 1.56361701],
[ 0.22300309, 0.58246951, 1.38180724],
[ 0.12363554, 0.28296928, 1.37532554],
[ 0.26255728, 0.74945806, 1.18056352],
[ 0.05584835, 0.13580419, 1.15735455],
[ 0.50075822, 0.73237569, 1.2520519 ],
[ 0.22529632, -0.03085884, 1.22130105],
[ 0.1752069 , 0.39383876, 1.56246206],
[ 0.24813409, 0.41115731, 1.61444763]], dtype=np.float32) / coordinate_scale
joints = {
'336': joints_336,
'377': joints_377,
'381': joints_381,
'384': joints_384,
'387': joints_387
}