sports2d 0.7.2__py3-none-any.whl → 0.8.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (17) hide show
  1. Sports2D/Demo/Config_demo.toml +12 -13
  2. Sports2D/Sports2D.py +27 -23
  3. Sports2D/Utilities/__init__.py +2 -1
  4. Sports2D/Utilities/common.py +2 -940
  5. Sports2D/Utilities/filter.py +2 -1
  6. Sports2D/Utilities/tests.py +40 -21
  7. Sports2D/__init__.py +2 -1
  8. Sports2D/process.py +540 -144
  9. {sports2d-0.7.2.dist-info → sports2d-0.8.0.dist-info}/METADATA +105 -83
  10. sports2d-0.8.0.dist-info/RECORD +15 -0
  11. {sports2d-0.7.2.dist-info → sports2d-0.8.0.dist-info}/WHEEL +1 -1
  12. sports2d-0.8.0.dist-info/entry_points.txt +3 -0
  13. Sports2D/Utilities/skeletons.py +0 -1000
  14. sports2d-0.7.2.dist-info/RECORD +0 -16
  15. sports2d-0.7.2.dist-info/entry_points.txt +0 -2
  16. {sports2d-0.7.2.dist-info → sports2d-0.8.0.dist-info/licenses}/LICENSE +0 -0
  17. {sports2d-0.7.2.dist-info → sports2d-0.8.0.dist-info}/top_level.txt +0 -0
Sports2D/Utilities/skeletons.py DELETED
@@ -1,1000 +0,0 @@
1
- #!/usr/bin/env python
2
- # -*- coding: utf-8 -*-
3
-
4
-
5
- '''
6
- ###########################################################################
7
- ## SKELETONS DEFINITIONS ##
8
- ###########################################################################
9
-
10
- The definition and hierarchy of the following skeletons are available:
11
- - RTMPose HALPE_26, COCO_133, COCO_133_WRIST, COCO_17, HAND, FACE, ANIMAL
12
- - OpenPose BODY_25B, BODY_25, BODY_135, COCO, MPII
13
- - Mediapipe BLAZEPOSE
14
- - AlphaPose HALPE_26, HALPE_68, HALPE_136, COCO_133, COCO, MPII
15
- (for COCO and MPII, AlphaPose must be run with the flag "--format cmu")
16
- - DeepLabCut CUSTOM: the skeleton will be defined in Config.toml
17
-
18
- N.B.: Not all face and hand keypoints are reported in the skeleton architecture,
19
- since some are redundant for the orientation of some bodies.
20
-
21
- N.B.: The corresponding OpenSim model files are provided in the "Pose2Sim/Empty project" folder.
22
- If you wish to use any other, you will need to adjust the markerset in the .osim model file,
23
- as well as in the scaling and IK setup files.
24
-
25
- Check the skeleton structure with:
26
- from anytree import Node, RenderTree
27
- for pre, _, node in RenderTree(model):
28
- print(f'{pre}{node.name} id={node.id}')
29
- '''
30
-
31
- ## INIT
32
- from anytree import Node
33
-
34
-
35
- ## AUTHORSHIP INFORMATION
36
- __author__ = "David Pagnon"
37
- __copyright__ = "Copyright 2021, Pose2Sim"
38
- __credits__ = ["David Pagnon"]
39
- __license__ = "BSD 3-Clause License"
40
- __version__ = "0.9.4"
41
- __maintainer__ = "David Pagnon"
42
- __email__ = "contact@david-pagnon.com"
43
- __status__ = "Development"
44
-
45
-
46
- '''HALPE_26 (full-body without hands, from AlphaPose, MMPose, etc.)
47
- https://github.com/MVIG-SJTU/AlphaPose/blob/master/docs/MODEL_ZOO.md
48
- https://github.com/open-mmlab/mmpose/tree/main/projects/rtmpose'''
49
- HALPE_26 = Node("Hip", id=19, children=[
50
- Node("RHip", id=12, children=[
51
- Node("RKnee", id=14, children=[
52
- Node("RAnkle", id=16, children=[
53
- Node("RBigToe", id=21, children=[
54
- Node("RSmallToe", id=23),
55
- ]),
56
- Node("RHeel", id=25),
57
- ]),
58
- ]),
59
- ]),
60
- Node("LHip", id=11, children=[
61
- Node("LKnee", id=13, children=[
62
- Node("LAnkle", id=15, children=[
63
- Node("LBigToe", id=20, children=[
64
- Node("LSmallToe", id=22),
65
- ]),
66
- Node("LHeel", id=24),
67
- ]),
68
- ]),
69
- ]),
70
- Node("Neck", id=18, children=[
71
- Node("Head", id=17, children=[
72
- Node("Nose", id=0),
73
- ]),
74
- Node("RShoulder", id=6, children=[
75
- Node("RElbow", id=8, children=[
76
- Node("RWrist", id=10),
77
- ]),
78
- ]),
79
- Node("LShoulder", id=5, children=[
80
- Node("LElbow", id=7, children=[
81
- Node("LWrist", id=9),
82
- ]),
83
- ]),
84
- ]),
85
- ])
86
-
87
-
88
- '''COCO_133_WRIST (full-body with hands and face, from AlphaPose, MMPose, etc.)
89
- https://github.com/MVIG-SJTU/AlphaPose/blob/master/docs/MODEL_ZOO.md
90
- https://github.com/open-mmlab/mmpose/tree/main/projects/rtmpose'''
91
- COCO_133_WRIST = Node("Hip", id=None, children=[
92
- Node("RHip", id=12, children=[
93
- Node("RKnee", id=14, children=[
94
- Node("RAnkle", id=16, children=[
95
- Node("RBigToe", id=20, children=[
96
- Node("RSmallToe", id=21),
97
- ]),
98
- Node("RHeel", id=22),
99
- ]),
100
- ]),
101
- ]),
102
- Node("LHip", id=11, children=[
103
- Node("LKnee", id=13, children=[
104
- Node("LAnkle", id=15, children=[
105
- Node("LBigToe", id=17, children=[
106
- Node("LSmallToe", id=18),
107
- ]),
108
- Node("LHeel", id=19),
109
- ]),
110
- ]),
111
- ]),
112
- Node("Neck", id=None, children=[
113
- Node("Nose", id=0, children=[
114
- Node("REye", id=2),
115
- Node("LEye", id=1),
116
- ]),
117
- Node("RShoulder", id=6, children=[
118
- Node("RElbow", id=8, children=[
119
- Node("RWrist", id=10, children=[
120
- Node("RThumb", id=114),
121
- Node("RIndex", id=117),
122
- Node("RPinky", id=129),
123
- ]),
124
- ]),
125
- ]),
126
- Node("LShoulder", id=5, children=[
127
- Node("LElbow", id=7, children=[
128
- Node("LWrist", id=9, children=[
129
- Node("LThumb", id=93),
130
- Node("LIndex", id=96),
131
- Node("LPinky", id=108),
132
- ])
133
- ]),
134
- ]),
135
- ]),
136
- ])
137
-
138
-
139
- '''COCO_133 (full-body with hands and face, from AlphaPose, MMPose, etc.)
140
- https://github.com/MVIG-SJTU/AlphaPose/blob/master/docs/MODEL_ZOO.md
141
- https://github.com/open-mmlab/mmpose/tree/main/projects/rtmpose'''
142
- COCO_133 = Node("Hip", id=None, children=[
143
- Node("RHip", id=12, children=[
144
- Node("RKnee", id=14, children=[
145
- Node("RAnkle", id=16, children=[
146
- Node("RBigToe", id=20, children=[
147
- Node("RSmallToe", id=21),
148
- ]),
149
- Node("RHeel", id=22),
150
- ]),
151
- ]),
152
- ]),
153
- Node("LHip", id=11, children=[
154
- Node("LKnee", id=13, children=[
155
- Node("LAnkle", id=15, children=[
156
- Node("LBigToe", id=17, children=[
157
- Node("LSmallToe", id=18),
158
- ]),
159
- Node("LHeel", id=19),
160
- ]),
161
- ]),
162
- ]),
163
- Node("Neck", id=None, children=[
164
- Node("Nose", id=0, children=[
165
- Node("REye", id=2),
166
- Node("LEye", id=1),
167
- Node("REar", id=4),
168
- Node("LEar", id=3),
169
- ]),
170
- Node("RShoulder", id=6, children=[
171
- Node("RElbow", id=8, children=[
172
- Node("RWrist", id=10, children=[
173
- Node("RThumb1", id=113, children=[
174
- Node("RThumb", id=114, children=[
175
- Node("RThumb3", id=115, children=[
176
- Node("RThumb4", id=116),
177
- ]),
178
- ]),
179
- ]),
180
- Node("RIndex", id=117, children=[
181
- Node("RIndex2", id=118, children=[
182
- Node("RIndex3", id=119, children=[
183
- Node("RIndex4", id=120),
184
- ]),
185
- ]),
186
- ]),
187
- Node("RMiddle1", id=121, children=[
188
- Node("RMiddle2", id=122, children=[
189
- Node("RMiddle3", id=123, children=[
190
- Node("RMiddle4", id=124),
191
- ]),
192
- ]),
193
- ]),
194
- Node("RRing1", id=125, children=[
195
- Node("RRing2", id=126, children=[
196
- Node("RRing3", id=127, children=[
197
- Node("RRing4", id=128),
198
- ]),
199
- ]),
200
- ]),
201
- Node("RPinky", id=129, children=[
202
- Node("RPinky2", id=130, children=[
203
- Node("RPinky3", id=131, children=[
204
- Node("RPinky4", id=132),
205
- ]),
206
- ]),
207
- ]),
208
- ]),
209
- ]),
210
- ]),
211
- Node("LShoulder", id=5, children=[
212
- Node("LElbow", id=7, children=[
213
- Node("LWrist", id=9, children=[
214
- Node("LThumb1", id=92, children=[
215
- Node("LThumb", id=93, children=[
216
- Node("LThumb3", id=94, children=[
217
- Node("LThumb4", id=95),
218
- ]),
219
- ]),
220
- ]),
221
- Node("LIndex", id=96, children=[
222
- Node("LIndex2", id=97, children=[
223
- Node("LIndex3", id=98, children=[
224
- Node("LIndex4", id=99),
225
- ]),
226
- ]),
227
- ]),
228
- Node("LMiddle1", id=100, children=[
229
- Node("LMiddle2", id=101, children=[
230
- Node("LMiddle3", id=102, children=[
231
- Node("LMiddle4", id=103),
232
- ]),
233
- ]),
234
- ]),
235
- Node("LRing1", id=104, children=[
236
- Node("LRing2", id=105, children=[
237
- Node("LRing3", id=106, children=[
238
- Node("LRing4", id=107),
239
- ]),
240
- ]),
241
- ]),
242
- Node("LPinky", id=108, children=[
243
- Node("LPinky2", id=109, children=[
244
- Node("LPinky3", id=110, children=[
245
- Node("LPinky4", id=111),
246
- ]),
247
- ]),
248
- ]),
249
- ])
250
- ]),
251
- ]),
252
- Node("Jaw1", id=23, children=[
253
- Node("Jaw2", id=24, children=[
254
- Node("Jaw3", id=25, children=[
255
- Node("Jaw4", id=26, children=[
256
- Node("Jaw5", id=27, children=[
257
- Node("Jaw6", id=28, children=[
258
- Node("Jaw7", id=29, children=[
259
- Node("Jaw8", id=30, children=[
260
- Node("Jaw9", id=31, children=[
261
- Node("Jaw10", id=32, children=[
262
- Node("Jaw11", id=33, children=[
263
- Node("Jaw12", id=34, children=[
264
- Node("Jaw13", id=35, children=[
265
- Node("Jaw14", id=36, children=[
266
- Node("Jaw15", id=37, children=[
267
- Node("Jaw16", id=38, children=[
268
- Node("Jaw17", id=39),
269
- ]),
270
- ]),
271
- ]),
272
- ]),
273
- ]),
274
- ]),
275
- ]),
276
- ]),
277
- ]),
278
- ]),
279
- ]),
280
- ]),
281
- ]),
282
- ]),
283
- ]),
284
- ]),
285
- Node("EyebrowR1", id=40, children=[
286
- Node("EyebrowR2", id=41, children=[
287
- Node("EyebrowR3", id=42, children=[
288
- Node("EyebrowR4", id=43, children=[
289
- Node("EyebrowR5", id=44, children=[
290
- Node("EyebrowL1", id=45, children=[
291
- Node("EyebrowL2", id=46, children=[
292
- Node("EyebrowL3", id=47, children=[
293
- Node("EyebrowL4", id=48, children=[
294
- Node("EyebrowL5", id=49)
295
- ]),
296
- ]),
297
- ]),
298
- ]),
299
- ]),
300
- ]),
301
- ]),
302
- ]),
303
- ]),
304
- Node("Nose1", id=50, children=[
305
- Node("Nose2", id=51, children=[
306
- Node("Nose3", id=52, children=[
307
- Node("Nose4", id=53, children=[
308
- Node("Nose5", id=54, children=[
309
- Node("Nose6", id=55, children=[
310
- Node("Nose7", id=56, children=[
311
- Node("Nose8", id=57, children=[
312
- Node("Nose9", id=58)
313
- ]),
314
- ]),
315
- ]),
316
- ]),
317
- ]),
318
- ]),
319
- ]),
320
- ]),
321
- Node("REye1", id=59, children=[
322
- Node("REye2", id=60, children=[
323
- Node("REye3", id=61, children=[
324
- Node("REye4", id=62, children=[
325
- Node("REye5", id=63, children=[
326
- Node("REye6", id=64)
327
- ]),
328
- ]),
329
- ]),
330
- ]),
331
- ]),
332
- Node("LEye1", id=65, children=[
333
- Node("LEye2", id=66, children=[
334
- Node("LEye3", id=67, children=[
335
- Node("LEye4", id=68, children=[
336
- Node("LEye5", id=69, children=[
337
- Node("LEye6", id=70)
338
- ]),
339
- ]),
340
- ]),
341
- ]),
342
- ]),
343
- Node("Mouth1", id=71, children=[
344
- Node("Mouth2", id=72, children=[
345
- Node("Mouth3", id=73, children=[
346
- Node("Mouth4", id=74, children=[
347
- Node("Mouth5", id=75, children=[
348
- Node("Mouth6", id=76, children=[
349
- Node("Mouth7", id=77, children=[
350
- Node("Mouth8", id=78, children=[
351
- Node("Mouth9", id=79, children=[
352
- Node("Mouth10", id=80, children=[
353
- Node("Mouth11", id=81, children=[
354
- Node("Mouth12", id=82, children=[
355
- Node("Mouth13", id=83, children=[
356
- Node("Mouth14", id=84, children=[
357
- Node("Mouth15", id=85, children=[
358
- Node("Mouth16", id=86, children=[
359
- Node("Mouth17", id=87, children=[
360
- Node("Mouth18", id=88, children=[
361
- Node("Mouth19", id=89, children=[
362
- Node("Mouth20", id=90)
363
- ]),
364
- ]),
365
- ]),
366
- ]),
367
- ]),
368
- ]),
369
- ]),
370
- ]),
371
- ]),
372
- ]),
373
- ]),
374
- ]),
375
- ]),
376
- ]),
377
- ]),
378
- ]),
379
- ]),
380
- ]),
381
- ]),
382
- ]),
383
- ])
384
-
385
-
386
- '''COCO_17 (full-body without hands and feet, from OpenPose, AlphaPose, OpenPifPaf, YOLO-pose, MMPose, etc.)
387
- https://github.com/open-mmlab/mmpose/tree/main/projects/rtmpose'''
388
- COCO_17 = Node("Hip", id=None, children=[
389
- Node("RHip", id=12, children=[
390
- Node("RKnee", id=14, children=[
391
- Node("RAnkle", id=16),
392
- ]),
393
- ]),
394
- Node("LHip", id=11, children=[
395
- Node("LKnee", id=13, children=[
396
- Node("LAnkle", id=15),
397
- ]),
398
- ]),
399
- Node("Neck", id=None, children=[
400
- Node("Nose", id=0),
401
- Node("RShoulder", id=6, children=[
402
- Node("RElbow", id=8, children=[
403
- Node("RWrist", id=10),
404
- ]),
405
- ]),
406
- Node("LShoulder", id=5, children=[
407
- Node("LElbow", id=7, children=[
408
- Node("LWrist", id=9),
409
- ]),
410
- ]),
411
- ]),
412
- ])
413
-
414
-
415
- '''HAND_21
416
- https://github.com/jin-s13/COCO-WholeBody/'''
417
- HAND_21 = Node("RWrist", id=0, children=[
418
- Node("RThumb1", id=1, children=[
419
- Node("RThumb2", id=2, children=[
420
- Node("RThumb3", id=3, children=[
421
- Node("RThumb4", id=4),
422
- ]),
423
- ]),
424
- ]),
425
- Node("RIndex1", id=5, children=[
426
- Node("RIndex2", id=6, children=[
427
- Node("RIndex3", id=7, children=[
428
- Node("RIndex4", id=8),
429
- ]),
430
- ]),
431
- ]),
432
- Node("RMiddle1", id=9, children=[
433
- Node("RMiddle2", id=10, children=[
434
- Node("RMiddle3", id=11, children=[
435
- Node("RMiddle4", id=12),
436
- ]),
437
- ]),
438
- ]),
439
- Node("RRing1", id=13, children=[
440
- Node("RRing2", id=14, children=[
441
- Node("RRing3", id=15, children=[
442
- Node("RRing4", id=16),
443
- ]),
444
- ]),
445
- ]),
446
- Node("RPinky1", id=17, children=[
447
- Node("RPinky2", id=18, children=[
448
- Node("RPinky3", id=19, children=[
449
- Node("RPinky4", id=20),
450
- ]),
451
- ]),
452
- ]),
453
- ])
454
-
455
-
456
- '''FACE_106
457
- https://github.com/jd-opensource/lapa-dataset'''
458
- FACE_106 = Node("root", id=None, children=[
459
- Node("Jaw0", id=0, children=[
460
- Node("Jaw1", id=1, children=[
461
- Node("Jaw2", id=2, children=[
462
- Node("Jaw3", id=3, children=[
463
- Node("Jaw4", id=4, children=[
464
- Node("Jaw5", id=5, children=[
465
- Node("Jaw6", id=6, children=[
466
- Node("Jaw7", id=7, children=[
467
- Node("Jaw8", id=8, children=[
468
- Node("Jaw9", id=9, children=[
469
- Node("Jaw10", id=10, children=[
470
- Node("Jaw11", id=11, children=[
471
- Node("Jaw12", id=12, children=[
472
- Node("Jaw13", id=13, children=[
473
- Node("Jaw14", id=14, children=[
474
- Node("Jaw15", id=15, children=[
475
- Node("Jaw16", id=16, children=[
476
- Node("Jaw17", id=17, children=[
477
- Node("Jaw18", id=18, children=[
478
- Node("Jaw19", id=19, children=[
479
- Node("Jaw20", id=20, children=[
480
- Node("Jaw21", id=21, children=[
481
- Node("Jaw22", id=22, children=[
482
- Node("Jaw23", id=23, children=[
483
- Node("Jaw24", id=24, children=[
484
- Node("Jaw25", id=25, children=[
485
- Node("Jaw26", id=26, children=[
486
- Node("Jaw27", id=27, children=[
487
- Node("Jaw28", id=28, children=[
488
- Node("Jaw29", id=29, children=[
489
- Node("Jaw30", id=30, children=[
490
- Node("Jaw31", id=31, children=[
491
- Node("Jaw32", id=32),
492
- ]),
493
- ]),
494
- ]),
495
- ]),
496
- ]),
497
- ]),
498
- ]),
499
- ]),
500
- ]),
501
- ]),
502
- ]),
503
- ]),
504
- ]),
505
- ]),
506
- ]),
507
- ]),
508
- ]),
509
- ]),
510
- ]),
511
- ]),
512
- ]),
513
- ]),
514
- ]),
515
- ]),
516
- ]),
517
- ]),
518
- ]),
519
- ]),
520
- ]),
521
- ]),
522
- ]),
523
- ]),
524
- Node("EyebrowR1", id=33, children=[
525
- Node("EyebrowR2", id=34, children=[
526
- Node("EyebrowR3", id=35, children=[
527
- Node("EyebrowR4", id=36, children=[
528
- Node("EyebrowR5", id=37, children=[
529
- Node("EyebrowR6", id=38, children=[
530
- Node("EyebrowR7", id=39, children=[
531
- Node("EyebrowR8", id=40, children=[
532
- Node("EyebrowR9", id=41)
533
- ]),
534
- ]),
535
- ]),
536
- ]),
537
- ]),
538
- ]),
539
- ]),
540
- ]),
541
- Node("EyebrowL1", id=42, children=[
542
- Node("EyebrowL2", id=43, children=[
543
- Node("EyebrowL3", id=44, children=[
544
- Node("EyebrowL4", id=45, children=[
545
- Node("EyebrowL5", id=46, children=[
546
- Node("EyebrowL6", id=47, children=[
547
- Node("EyebrowL7", id=48, children=[
548
- Node("EyebrowL8", id=49, children=[
549
- Node("EyebrowL9", id=50)
550
- ]),
551
- ]),
552
- ]),
553
- ]),
554
- ]),
555
- ]),
556
- ]),
557
- Node("Nose1", id=51, children=[
558
- Node("Nose2", id=52, children=[
559
- Node("Nose3", id=53, children=[
560
- Node("Nose4", id=54, children=[
561
- Node("Nose5", id=55, children=[
562
- Node("Nose6", id=56, children=[
563
- Node("Nose7", id=57, children=[
564
- Node("Nose8", id=58, children=[
565
- Node("Nose9", id=59)
566
- ]),
567
- ]),
568
- ]),
569
- ]),
570
- ]),
571
- ]),
572
- ]),
573
- ]),
574
- Node("REye1", id=60, children=[
575
- Node("REye2", id=61, children=[
576
- Node("REye3", id=62, children=[
577
- Node("REye4", id=63, children=[
578
- Node("REye5", id=64, children=[
579
- Node("REye6", id=65, children=[
580
- Node("REye7", id=66, children=[
581
- Node("REye8", id=67)
582
- ]),
583
- ]),
584
- ]),
585
- ]),
586
- ]),
587
- ]),
588
- ]),
589
- Node("LEye1", id=68, children=[
590
- Node("LEye2", id=69, children=[
591
- Node("LEye3", id=70, children=[
592
- Node("LEye4", id=71, children=[
593
- Node("LEye5", id=72, children=[
594
- Node("LEye6", id=73, children=[
595
- Node("LEye7", id=74, children=[
596
- Node("LEye8", id=75)
597
- ]),
598
- ]),
599
- ]),
600
- ]),
601
- ]),
602
- ]),
603
- ]),
604
- Node("Mouth1", id=76, children=[
605
- Node("Mouth2", id=77, children=[
606
- Node("Mouth3", id=78, children=[
607
- Node("Mouth4", id=79, children=[
608
- Node("Mouth5", id=80, children=[
609
- Node("Mouth6", id=81, children=[
610
- Node("Mouth7", id=82, children=[
611
- Node("Mouth8", id=83, children=[
612
- Node("Mouth9", id=84, children=[
613
- Node("Mouth10", id=85, children=[
614
- Node("Mouth11", id=86, children=[
615
- Node("Mouth12", id=87, children=[
616
- Node("Mouth13", id=88, children=[
617
- Node("Mouth14", id=89, children=[
618
- Node("Mouth15", id=90, children=[
619
- Node("Mouth16", id=91, children=[
620
- Node("Mouth17", id=92, children=[
621
- Node("Mouth18", id=93, children=[
622
- Node("Mouth19", id=94, children=[
623
- Node("Mouth20", id=95)
624
- ]),
625
- ]),
626
- ]),
627
- ]),
628
- ]),
629
- ]),
630
- ]),
631
- ]),
632
- ]),
633
- ]),
634
- ]),
635
- ]),
636
- ]),
637
- ]),
638
- ]),
639
- ]),
640
- ]),
641
- ]),
642
- ]),
643
- ]),
644
- ])
645
-
646
- '''ANIMAL2D_17 (full-body animal)
647
- https://github.com/AlexTheBad/AP-10K/'''
648
- ANIMAL2D_17 = Node("Hip", id=4, children=[
649
- Node("RHip", id=14, children=[
650
- Node("RKnee", id=15, children=[
651
- Node("RAnkle", id=16),
652
- ]),
653
- ]),
654
- Node("LHip", id=11, children=[
655
- Node("LKnee", id=12, children=[
656
- Node("LAnkle", id=13),
657
- ]),
658
- ]),
659
- Node("Neck", id=3, children=[
660
- Node("Nose", id=2, children=[
661
- Node("REye", id=1),
662
- Node("LEye", id=0),
663
- ]),
664
- Node("RShoulder", id=8, children=[
665
- Node("RElbow", id=9, children=[
666
- Node("RWrist", id=10),
667
- ]),
668
- ]),
669
- Node("LShoulder", id=5, children=[
670
- Node("LElbow", id=6, children=[
671
- Node("LWrist", id=7),
672
- ]),
673
- ]),
674
- ]),
675
- ])
676
-
677
-
678
- '''BODY_25B (full-body without hands, experimental, from OpenPose)
679
- https://github.com/CMU-Perceptual-Computing-Lab/openpose_train/blob/master/experimental_models/README.md'''
680
- BODY_25B = Node("CHip", id=None, children=[
681
- Node("RHip", id=12, children=[
682
- Node("RKnee", id=14, children=[
683
- Node("RAnkle", id=16, children=[
684
- Node("RBigToe", id=22, children=[
685
- Node("RSmallToe", id=23),
686
- ]),
687
- Node("RHeel", id=24),
688
- ]),
689
- ]),
690
- ]),
691
- Node("LHip", id=11, children=[
692
- Node("LKnee", id=13, children=[
693
- Node("LAnkle", id=15, children=[
694
- Node("LBigToe", id=19, children=[
695
- Node("LSmallToe", id=20),
696
- ]),
697
- Node("LHeel", id=21),
698
- ]),
699
- ]),
700
- ]),
701
- Node("Neck", id=17, children=[
702
- Node("Head", id=18, children=[
703
- Node("Nose", id=0),
704
- ]),
705
- Node("RShoulder", id=6, children=[
706
- Node("RElbow", id=8, children=[
707
- Node("RWrist", id=10),
708
- ]),
709
- ]),
710
- Node("LShoulder", id=5, children=[
711
- Node("LElbow", id=7, children=[
712
- Node("LWrist", id=9),
713
- ]),
714
- ]),
715
- ]),
716
- ])
717
-
718
-
719
- '''BODY_25 (full-body without hands, standard, from OpenPose)
720
- https://github.com/CMU-Perceptual-Computing-Lab/openpose/tree/master/models'''
721
- BODY_25 = Node("CHip", id=8, children=[
722
- Node("RHip", id=9, children=[
723
- Node("RKnee", id=10, children=[
724
- Node("RAnkle", id=11, children=[
725
- Node("RBigToe", id=22, children=[
726
- Node("RSmallToe", id=23),
727
- ]),
728
- Node("RHeel", id=24),
729
- ]),
730
- ]),
731
- ]),
732
- Node("LHip", id=12, children=[
733
- Node("LKnee", id=13, children=[
734
- Node("LAnkle", id=14, children=[
735
- Node("LBigToe", id=19, children=[
736
- Node("LSmallToe", id=20),
737
- ]),
738
- Node("LHeel", id=21),
739
- ]),
740
- ]),
741
- ]),
742
- Node("Neck", id=1, children=[
743
- Node("Nose", id=0),
744
- Node("RShoulder", id=2, children=[
745
- Node("RElbow", id=3, children=[
746
- Node("RWrist", id=4),
747
- ]),
748
- ]),
749
- Node("LShoulder", id=5, children=[
750
- Node("LElbow", id=6, children=[
751
- Node("LWrist", id=7),
752
- ]),
753
- ]),
754
- ]),
755
- ])
756
-
757
-
758
- '''BODY_135 (full-body with hands and face, experimental, from OpenPose)
759
- https://github.com/CMU-Perceptual-Computing-Lab/openpose_train/blob/master/experimental_models/README.md)'''
760
- BODY_135 = Node("CHip", id=None, children=[
761
- Node("RHip", id=12, children=[
762
- Node("RKnee", id=14, children=[
763
- Node("RAnkle", id=16, children=[
764
- Node("RBigToe", id=22, children=[
765
- Node("RSmallToe", id=23),
766
- ]),
767
- Node("RHeel", id=24),
768
- ]),
769
- ]),
770
- ]),
771
- Node("LHip", id=11, children=[
772
- Node("LKnee", id=13, children=[
773
- Node("LAnkle", id=15, children=[
774
- Node("LBigToe", id=19, children=[
775
- Node("LSmallToe", id=20),
776
- ]),
777
- Node("LHeel", id=21),
778
- ]),
779
- ]),
780
- ]),
781
- Node("Neck", id=17, children=[
782
- Node("Head", id=18, children=[
783
- Node("Nose", id=0),
784
- ]),
785
- Node("RShoulder", id=6, children=[
786
- Node("RElbow", id=8, children=[
787
- Node("RWrist", id=10, children=[
788
- Node("RThumb", id=48),
789
- Node("RIndex", id=51),
790
- Node("RPinky", id=63),
791
- ]),
792
- ]),
793
- ]),
794
- Node("LShoulder", id=5, children=[
795
- Node("LElbow", id=7, children=[
796
- Node("LWrist", id=9, children=[
797
- Node("LThumb", id=27),
798
- Node("LIndex", id=30),
799
- Node("LPinky", id=42),
800
- ]),
801
- ]),
802
- ]),
803
- ]),
804
- ])
805
-
806
-
807
- '''BLAZEPOSE (full-body with simplified hand and foot, from mediapipe)
808
- https://google.github.io/mediapipe/solutions/pose'''
809
- BLAZEPOSE = Node("root", id=None, children=[
810
- Node("right_hip", id=24, children=[
811
- Node("right_knee", id=26, children=[
812
- Node("right_ankle", id=28, children=[
813
- Node("right_heel", id=30),
814
- Node("right_foot_index", id=32),
815
- ]),
816
- ]),
817
- ]),
818
- Node("left_hip", id=23, children=[
819
- Node("left_knee", id=25, children=[
820
- Node("left_ankle", id=27, children=[
821
- Node("left_heel", id=29),
822
- Node("left_foot_index", id=31),
823
- ]),
824
- ]),
825
- ]),
826
- Node("nose", id=0, children=[
827
- Node("right_eye", id=5),
828
- Node("left_eye", id=2),
829
- ]),
830
- Node("right_shoulder", id=12, children=[
831
- Node("right_elbow", id=14, children=[
832
- Node("right_wrist", id=16, children=[
833
- Node("right_pinky", id=18),
834
- Node("right_index", id=20),
835
- Node("right_thumb", id=22),
836
- ]),
837
- ]),
838
- ]),
839
- Node("left_shoulder", id=11, children=[
840
- Node("left_elbow", id=13, children=[
841
- Node("left_wrist", id=15, children=[
842
- Node("left_pinky", id=17),
843
- Node("left_index", id=19),
844
- Node("left_thumb", id=21),
845
- ]),
846
- ]),
847
- ]),
848
- ])
849
-
850
-
851
- '''HALPE_68 (full-body with hands without face, from AlphaPose, MMPose, etc.)
852
- https://github.com/MVIG-SJTU/AlphaPose/blob/master/docs/MODEL_ZOO.md'''
853
- HALPE_68 = Node("Hip", id=19, children=[
854
- Node("RHip", id=12, children=[
855
- Node("RKnee", id=14, children=[
856
- Node("RAnkle", id=16, children=[
857
- Node("RBigToe", id=21, children=[
858
- Node("RSmallToe", id=23),
859
- ]),
860
- Node("RHeel", id=25),
861
- ]),
862
- ]),
863
- ]),
864
- Node("LHip", id=11, children=[
865
- Node("LKnee", id=13, children=[
866
- Node("LAnkle", id=15, children=[
867
- Node("LBigToe", id=20, children=[
868
- Node("LSmallToe", id=22),
869
- ]),
870
- Node("LHeel", id=24),
871
- ]),
872
- ]),
873
- ]),
874
- Node("Neck", id=18, children=[
875
- Node("Nose", id=0),
876
- Node("RShoulder", id=6, children=[
877
- Node("RElbow", id=8, children=[
878
- Node("RWrist", id=10, children=[
879
- Node("RThumb", id=49),
880
- Node("RIndex", id=52),
881
- Node("RPinky", id=64),
882
- ]),
883
- ]),
884
- ]),
885
- Node("LShoulder", id=5, children=[
886
- Node("LElbow", id=7, children=[
887
- Node("LWrist", id=9, children=[
888
- Node("LThumb", id=28),
889
- Node("LIndex", id=31),
890
- Node("LPinky", id=43),
891
- ])
892
- ]),
893
- ]),
894
- ]),
895
- ])
896
-
897
-
898
- '''HALPE_136 (full-body with hands and face, from AlphaPose, MMPose, etc.)
899
- https://github.com/MVIG-SJTU/AlphaPose/blob/master/docs/MODEL_ZOO.md'''
900
- HALPE_136 = Node("Hip", id=19, children=[
901
- Node("RHip", id=12, children=[
902
- Node("RKnee", id=14, children=[
903
- Node("RAnkle", id=16, children=[
904
- Node("RBigToe", id=21, children=[
905
- Node("RSmallToe", id=23),
906
- ]),
907
- Node("RHeel", id=25),
908
- ]),
909
- ]),
910
- ]),
911
- Node("LHip", id=11, children=[
912
- Node("LKnee", id=13, children=[
913
- Node("LAnkle", id=15, children=[
914
- Node("LBigToe", id=20, children=[
915
- Node("LSmallToe", id=22),
916
- ]),
917
- Node("LHeel", id=24),
918
- ]),
919
- ]),
920
- ]),
921
- Node("Neck", id=18, children=[
922
- Node("Nose", id=0),
923
- Node("RShoulder", id=6, children=[
924
- Node("RElbow", id=8, children=[
925
- Node("RWrist", id=10, children=[
926
- Node("RThumb", id=117),
927
- Node("RIndex", id=120),
928
- Node("RPinky", id=132),
929
- ]),
930
- ]),
931
- ]),
932
- Node("LShoulder", id=5, children=[
933
- Node("LElbow", id=7, children=[
934
- Node("LWrist", id=9, children=[
935
- Node("LThumb", id=96),
936
- Node("LIndex", id=99),
937
- Node("LPinky", id=111),
938
- ])
939
- ]),
940
- ]),
941
- ]),
942
- ])
943
-
944
-
945
- '''COCO (full-body without hands and feet, from OpenPose, AlphaPose, OpenPifPaf, YOLO-pose, MMPose, etc.)
946
- https://github.com/CMU-Perceptual-Computing-Lab/openpose/tree/master/models'''
947
- COCO = Node("CHip", id=None, children=[
948
- Node("RHip", id=8, children=[
949
- Node("RKnee", id=9, children=[
950
- Node("RAnkle", id=10),
951
- ]),
952
- ]),
953
- Node("LHip", id=11, children=[
954
- Node("LKnee", id=12, children=[
955
- Node("LAnkle", id=13),
956
- ]),
957
- ]),
958
- Node("Neck", id=1, children=[
959
- Node("Nose", id=0),
960
- Node("RShoulder", id=2, children=[
961
- Node("RElbow", id=3, children=[
962
- Node("RWrist", id=4),
963
- ]),
964
- ]),
965
- Node("LShoulder", id=5, children=[
966
- Node("LElbow", id=6, children=[
967
- Node("LWrist", id=7),
968
- ]),
969
- ]),
970
- ]),
971
- ])
972
-
973
-
974
- '''MPII (full-body without hands and feet, from OpenPose, AlphaPose, OpenPifPaf, YOLO-pose, MMPose, etc.)
975
- https://github.com/CMU-Perceptual-Computing-Lab/openpose/tree/master/models'''
976
- MPII = Node("CHip", id=14, children=[
977
- Node("RHip", id=8, children=[
978
- Node("RKnee", id=9, children=[
979
- Node("RAnkle", id=10),
980
- ]),
981
- ]),
982
- Node("LHip", id=11, children=[
983
- Node("LKnee", id=12, children=[
984
- Node("LAnkle", id=13),
985
- ]),
986
- ]),
987
- Node("Neck", id=1, children=[
988
- Node("Nose", id=0),
989
- Node("RShoulder", id=2, children=[
990
- Node("RElbow", id=3, children=[
991
- Node("RWrist", id=4),
992
- ]),
993
- ]),
994
- Node("LShoulder", id=5, children=[
995
- Node("LElbow", id=6, children=[
996
- Node("LWrist", id=7),
997
- ]),
998
- ]),
999
- ]),
1000
- ])