Spaces:
Running
Running
fiesty-bear
commited on
added skeleton
Browse files
config.py
CHANGED
@@ -78,3 +78,249 @@ LABELS_TO_IDS = {
|
|
78 |
"Upper Teeth": 26,
|
79 |
"Tongue": 27,
|
80 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
78 |
"Upper Teeth": 26,
|
79 |
"Tongue": 27,
|
80 |
}
|
81 |
+
|
82 |
+
skeleton_info={
|
83 |
+
0:
|
84 |
+
dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
|
85 |
+
1:
|
86 |
+
dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
|
87 |
+
2:
|
88 |
+
dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
|
89 |
+
3:
|
90 |
+
dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
|
91 |
+
4:
|
92 |
+
dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
|
93 |
+
5:
|
94 |
+
dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
|
95 |
+
6:
|
96 |
+
dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
|
97 |
+
7:
|
98 |
+
dict(
|
99 |
+
link=('left_shoulder', 'right_shoulder'),
|
100 |
+
id=7,
|
101 |
+
color=[51, 153, 255]),
|
102 |
+
8:
|
103 |
+
dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
|
104 |
+
9:
|
105 |
+
dict(
|
106 |
+
link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
|
107 |
+
10:
|
108 |
+
dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
|
109 |
+
11:
|
110 |
+
dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
|
111 |
+
12:
|
112 |
+
dict(link=('left_eye', 'right_eye'), id=12, color=[51, 153, 255]),
|
113 |
+
13:
|
114 |
+
dict(link=('nose', 'left_eye'), id=13, color=[51, 153, 255]),
|
115 |
+
14:
|
116 |
+
dict(link=('nose', 'right_eye'), id=14, color=[51, 153, 255]),
|
117 |
+
15:
|
118 |
+
dict(link=('left_eye', 'left_ear'), id=15, color=[51, 153, 255]),
|
119 |
+
16:
|
120 |
+
dict(link=('right_eye', 'right_ear'), id=16, color=[51, 153, 255]),
|
121 |
+
17:
|
122 |
+
dict(link=('left_ear', 'left_shoulder'), id=17, color=[51, 153, 255]),
|
123 |
+
18:
|
124 |
+
dict(
|
125 |
+
link=('right_ear', 'right_shoulder'), id=18, color=[51, 153, 255]),
|
126 |
+
19:
|
127 |
+
dict(link=('left_ankle', 'left_big_toe'), id=19, color=[0, 255, 0]),
|
128 |
+
20:
|
129 |
+
dict(link=('left_ankle', 'left_small_toe'), id=20, color=[0, 255, 0]),
|
130 |
+
21:
|
131 |
+
dict(link=('left_ankle', 'left_heel'), id=21, color=[0, 255, 0]),
|
132 |
+
22:
|
133 |
+
dict(
|
134 |
+
link=('right_ankle', 'right_big_toe'), id=22, color=[255, 128, 0]),
|
135 |
+
23:
|
136 |
+
dict(
|
137 |
+
link=('right_ankle', 'right_small_toe'),
|
138 |
+
id=23,
|
139 |
+
color=[255, 128, 0]),
|
140 |
+
24:
|
141 |
+
dict(link=('right_ankle', 'right_heel'), id=24, color=[255, 128, 0]),
|
142 |
+
25:
|
143 |
+
dict(
|
144 |
+
link=('left_hand_root', 'left_thumb1'), id=25, color=[255, 128,
|
145 |
+
0]),
|
146 |
+
26:
|
147 |
+
dict(link=('left_thumb1', 'left_thumb2'), id=26, color=[255, 128, 0]),
|
148 |
+
27:
|
149 |
+
dict(link=('left_thumb2', 'left_thumb3'), id=27, color=[255, 128, 0]),
|
150 |
+
28:
|
151 |
+
dict(link=('left_thumb3', 'left_thumb4'), id=28, color=[255, 128, 0]),
|
152 |
+
29:
|
153 |
+
dict(
|
154 |
+
link=('left_hand_root', 'left_forefinger1'),
|
155 |
+
id=29,
|
156 |
+
color=[255, 153, 255]),
|
157 |
+
30:
|
158 |
+
dict(
|
159 |
+
link=('left_forefinger1', 'left_forefinger2'),
|
160 |
+
id=30,
|
161 |
+
color=[255, 153, 255]),
|
162 |
+
31:
|
163 |
+
dict(
|
164 |
+
link=('left_forefinger2', 'left_forefinger3'),
|
165 |
+
id=31,
|
166 |
+
color=[255, 153, 255]),
|
167 |
+
32:
|
168 |
+
dict(
|
169 |
+
link=('left_forefinger3', 'left_forefinger4'),
|
170 |
+
id=32,
|
171 |
+
color=[255, 153, 255]),
|
172 |
+
33:
|
173 |
+
dict(
|
174 |
+
link=('left_hand_root', 'left_middle_finger1'),
|
175 |
+
id=33,
|
176 |
+
color=[102, 178, 255]),
|
177 |
+
34:
|
178 |
+
dict(
|
179 |
+
link=('left_middle_finger1', 'left_middle_finger2'),
|
180 |
+
id=34,
|
181 |
+
color=[102, 178, 255]),
|
182 |
+
35:
|
183 |
+
dict(
|
184 |
+
link=('left_middle_finger2', 'left_middle_finger3'),
|
185 |
+
id=35,
|
186 |
+
color=[102, 178, 255]),
|
187 |
+
36:
|
188 |
+
dict(
|
189 |
+
link=('left_middle_finger3', 'left_middle_finger4'),
|
190 |
+
id=36,
|
191 |
+
color=[102, 178, 255]),
|
192 |
+
37:
|
193 |
+
dict(
|
194 |
+
link=('left_hand_root', 'left_ring_finger1'),
|
195 |
+
id=37,
|
196 |
+
color=[255, 51, 51]),
|
197 |
+
38:
|
198 |
+
dict(
|
199 |
+
link=('left_ring_finger1', 'left_ring_finger2'),
|
200 |
+
id=38,
|
201 |
+
color=[255, 51, 51]),
|
202 |
+
39:
|
203 |
+
dict(
|
204 |
+
link=('left_ring_finger2', 'left_ring_finger3'),
|
205 |
+
id=39,
|
206 |
+
color=[255, 51, 51]),
|
207 |
+
40:
|
208 |
+
dict(
|
209 |
+
link=('left_ring_finger3', 'left_ring_finger4'),
|
210 |
+
id=40,
|
211 |
+
color=[255, 51, 51]),
|
212 |
+
41:
|
213 |
+
dict(
|
214 |
+
link=('left_hand_root', 'left_pinky_finger1'),
|
215 |
+
id=41,
|
216 |
+
color=[0, 255, 0]),
|
217 |
+
42:
|
218 |
+
dict(
|
219 |
+
link=('left_pinky_finger1', 'left_pinky_finger2'),
|
220 |
+
id=42,
|
221 |
+
color=[0, 255, 0]),
|
222 |
+
43:
|
223 |
+
dict(
|
224 |
+
link=('left_pinky_finger2', 'left_pinky_finger3'),
|
225 |
+
id=43,
|
226 |
+
color=[0, 255, 0]),
|
227 |
+
44:
|
228 |
+
dict(
|
229 |
+
link=('left_pinky_finger3', 'left_pinky_finger4'),
|
230 |
+
id=44,
|
231 |
+
color=[0, 255, 0]),
|
232 |
+
45:
|
233 |
+
dict(
|
234 |
+
link=('right_hand_root', 'right_thumb1'),
|
235 |
+
id=45,
|
236 |
+
color=[255, 128, 0]),
|
237 |
+
46:
|
238 |
+
dict(
|
239 |
+
link=('right_thumb1', 'right_thumb2'), id=46, color=[255, 128, 0]),
|
240 |
+
47:
|
241 |
+
dict(
|
242 |
+
link=('right_thumb2', 'right_thumb3'), id=47, color=[255, 128, 0]),
|
243 |
+
48:
|
244 |
+
dict(
|
245 |
+
link=('right_thumb3', 'right_thumb4'), id=48, color=[255, 128, 0]),
|
246 |
+
49:
|
247 |
+
dict(
|
248 |
+
link=('right_hand_root', 'right_forefinger1'),
|
249 |
+
id=49,
|
250 |
+
color=[255, 153, 255]),
|
251 |
+
50:
|
252 |
+
dict(
|
253 |
+
link=('right_forefinger1', 'right_forefinger2'),
|
254 |
+
id=50,
|
255 |
+
color=[255, 153, 255]),
|
256 |
+
51:
|
257 |
+
dict(
|
258 |
+
link=('right_forefinger2', 'right_forefinger3'),
|
259 |
+
id=51,
|
260 |
+
color=[255, 153, 255]),
|
261 |
+
52:
|
262 |
+
dict(
|
263 |
+
link=('right_forefinger3', 'right_forefinger4'),
|
264 |
+
id=52,
|
265 |
+
color=[255, 153, 255]),
|
266 |
+
53:
|
267 |
+
dict(
|
268 |
+
link=('right_hand_root', 'right_middle_finger1'),
|
269 |
+
id=53,
|
270 |
+
color=[102, 178, 255]),
|
271 |
+
54:
|
272 |
+
dict(
|
273 |
+
link=('right_middle_finger1', 'right_middle_finger2'),
|
274 |
+
id=54,
|
275 |
+
color=[102, 178, 255]),
|
276 |
+
55:
|
277 |
+
dict(
|
278 |
+
link=('right_middle_finger2', 'right_middle_finger3'),
|
279 |
+
id=55,
|
280 |
+
color=[102, 178, 255]),
|
281 |
+
56:
|
282 |
+
dict(
|
283 |
+
link=('right_middle_finger3', 'right_middle_finger4'),
|
284 |
+
id=56,
|
285 |
+
color=[102, 178, 255]),
|
286 |
+
57:
|
287 |
+
dict(
|
288 |
+
link=('right_hand_root', 'right_ring_finger1'),
|
289 |
+
id=57,
|
290 |
+
color=[255, 51, 51]),
|
291 |
+
58:
|
292 |
+
dict(
|
293 |
+
link=('right_ring_finger1', 'right_ring_finger2'),
|
294 |
+
id=58,
|
295 |
+
color=[255, 51, 51]),
|
296 |
+
59:
|
297 |
+
dict(
|
298 |
+
link=('right_ring_finger2', 'right_ring_finger3'),
|
299 |
+
id=59,
|
300 |
+
color=[255, 51, 51]),
|
301 |
+
60:
|
302 |
+
dict(
|
303 |
+
link=('right_ring_finger3', 'right_ring_finger4'),
|
304 |
+
id=60,
|
305 |
+
color=[255, 51, 51]),
|
306 |
+
61:
|
307 |
+
dict(
|
308 |
+
link=('right_hand_root', 'right_pinky_finger1'),
|
309 |
+
id=61,
|
310 |
+
color=[0, 255, 0]),
|
311 |
+
62:
|
312 |
+
dict(
|
313 |
+
link=('right_pinky_finger1', 'right_pinky_finger2'),
|
314 |
+
id=62,
|
315 |
+
color=[0, 255, 0]),
|
316 |
+
63:
|
317 |
+
dict(
|
318 |
+
link=('right_pinky_finger2', 'right_pinky_finger3'),
|
319 |
+
id=63,
|
320 |
+
color=[0, 255, 0]),
|
321 |
+
64:
|
322 |
+
dict(
|
323 |
+
link=('right_pinky_finger3', 'right_pinky_finger4'),
|
324 |
+
id=64,
|
325 |
+
color=[0, 255, 0])
|
326 |
+
},
|