Skip to content

GitLab

Commit ca8a762a authored by chenzk's avatar chenzk
Browse files

v1.0

parents
Pipeline #716 failed with stages
in 0 seconds
Hide whitespace changes
Inline Side-by-side
Showing with 6137 additions and 0 deletions
configs/_base_/datasets/freihand2d.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='freihand',
paper_info=dict(
author='Zimmermann, Christian and Ceylan, Duygu and '
'Yang, Jimei and Russell, Bryan and '
'Argus, Max and Brox, Thomas',
title='Freihand: A dataset for markerless capture of hand pose '
'and shape from single rgb images',
container='Proceedings of the IEEE International '
'Conference on Computer Vision',
year='2019',
homepage='https://lmb.informatik.uni-freiburg.de/projects/freihand/',
),
keypoint_info={
0:
dict(name='wrist', id=0, color=[255, 255, 255], type='', swap=''),
1:
dict(name='thumb1', id=1, color=[255, 128, 0], type='', swap=''),
2:
dict(name='thumb2', id=2, color=[255, 128, 0], type='', swap=''),
3:
dict(name='thumb3', id=3, color=[255, 128, 0], type='', swap=''),
4:
dict(name='thumb4', id=4, color=[255, 128, 0], type='', swap=''),
5:
dict(
name='forefinger1', id=5, color=[255, 153, 255], type='', swap=''),
6:
dict(
name='forefinger2', id=6, color=[255, 153, 255], type='', swap=''),
7:
dict(
name='forefinger3', id=7, color=[255, 153, 255], type='', swap=''),
8:
dict(
name='forefinger4', id=8, color=[255, 153, 255], type='', swap=''),
9:
dict(
name='middle_finger1',
id=9,
color=[102, 178, 255],
type='',
swap=''),
10:
dict(
name='middle_finger2',
id=10,
color=[102, 178, 255],
type='',
swap=''),
11:
dict(
name='middle_finger3',
id=11,
color=[102, 178, 255],
type='',
swap=''),
12:
dict(
name='middle_finger4',
id=12,
color=[102, 178, 255],
type='',
swap=''),
13:
dict(
name='ring_finger1', id=13, color=[255, 51, 51], type='', swap=''),
14:
dict(
name='ring_finger2', id=14, color=[255, 51, 51], type='', swap=''),
15:
dict(
name='ring_finger3', id=15, color=[255, 51, 51], type='', swap=''),
16:
dict(
name='ring_finger4', id=16, color=[255, 51, 51], type='', swap=''),
17:
dict(name='pinky_finger1', id=17, color=[0, 255, 0], type='', swap=''),
18:
dict(name='pinky_finger2', id=18, color=[0, 255, 0], type='', swap=''),
19:
dict(name='pinky_finger3', id=19, color=[0, 255, 0], type='', swap=''),
20:
dict(name='pinky_finger4', id=20, color=[0, 255, 0], type='', swap='')
},
skeleton_info={
0:
dict(link=('wrist', 'thumb1'), id=0, color=[255, 128, 0]),
1:
dict(link=('thumb1', 'thumb2'), id=1, color=[255, 128, 0]),
2:
dict(link=('thumb2', 'thumb3'), id=2, color=[255, 128, 0]),
3:
dict(link=('thumb3', 'thumb4'), id=3, color=[255, 128, 0]),
4:
dict(link=('wrist', 'forefinger1'), id=4, color=[255, 153, 255]),
5:
dict(link=('forefinger1', 'forefinger2'), id=5, color=[255, 153, 255]),
6:
dict(link=('forefinger2', 'forefinger3'), id=6, color=[255, 153, 255]),
7:
dict(link=('forefinger3', 'forefinger4'), id=7, color=[255, 153, 255]),
8:
dict(link=('wrist', 'middle_finger1'), id=8, color=[102, 178, 255]),
9:
dict(
link=('middle_finger1', 'middle_finger2'),
id=9,
color=[102, 178, 255]),
10:
dict(
link=('middle_finger2', 'middle_finger3'),
id=10,
color=[102, 178, 255]),
11:
dict(
link=('middle_finger3', 'middle_finger4'),
id=11,
color=[102, 178, 255]),
12:
dict(link=('wrist', 'ring_finger1'), id=12, color=[255, 51, 51]),
13:
dict(
link=('ring_finger1', 'ring_finger2'), id=13, color=[255, 51, 51]),
14:
dict(
link=('ring_finger2', 'ring_finger3'), id=14, color=[255, 51, 51]),
15:
dict(
link=('ring_finger3', 'ring_finger4'), id=15, color=[255, 51, 51]),
16:
dict(link=('wrist', 'pinky_finger1'), id=16, color=[0, 255, 0]),
17:
dict(
link=('pinky_finger1', 'pinky_finger2'), id=17, color=[0, 255, 0]),
18:
dict(
link=('pinky_finger2', 'pinky_finger3'), id=18, color=[0, 255, 0]),
19:
dict(
link=('pinky_finger3', 'pinky_finger4'), id=19, color=[0, 255, 0])
},
joint_weights=[1.] * 21,
sigmas=[])
configs/_base_/datasets/h36m.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='h36m',
paper_info=dict(
author='Ionescu, Catalin and Papava, Dragos and '
'Olaru, Vlad and Sminchisescu, Cristian',
title='Human3.6M: Large Scale Datasets and Predictive '
'Methods for 3D Human Sensing in Natural Environments',
container='IEEE Transactions on Pattern Analysis and '
'Machine Intelligence',
year='2014',
homepage='http://vision.imar.ro/human3.6m/description.php',
),
keypoint_info={
0:
dict(name='root', id=0, color=[51, 153, 255], type='lower', swap=''),
1:
dict(
name='right_hip',
id=1,
color=[255, 128, 0],
type='lower',
swap='left_hip'),
2:
dict(
name='right_knee',
id=2,
color=[255, 128, 0],
type='lower',
swap='left_knee'),
3:
dict(
name='right_foot',
id=3,
color=[255, 128, 0],
type='lower',
swap='left_foot'),
4:
dict(
name='left_hip',
id=4,
color=[0, 255, 0],
type='lower',
swap='right_hip'),
5:
dict(
name='left_knee',
id=5,
color=[0, 255, 0],
type='lower',
swap='right_knee'),
6:
dict(
name='left_foot',
id=6,
color=[0, 255, 0],
type='lower',
swap='right_foot'),
7:
dict(name='spine', id=7, color=[51, 153, 255], type='upper', swap=''),
8:
dict(name='thorax', id=8, color=[51, 153, 255], type='upper', swap=''),
9:
dict(
name='neck_base',
id=9,
color=[51, 153, 255],
type='upper',
swap=''),
10:
dict(name='head', id=10, color=[51, 153, 255], type='upper', swap=''),
11:
dict(
name='left_shoulder',
id=11,
color=[0, 255, 0],
type='upper',
swap='right_shoulder'),
12:
dict(
name='left_elbow',
id=12,
color=[0, 255, 0],
type='upper',
swap='right_elbow'),
13:
dict(
name='left_wrist',
id=13,
color=[0, 255, 0],
type='upper',
swap='right_wrist'),
14:
dict(
name='right_shoulder',
id=14,
color=[255, 128, 0],
type='upper',
swap='left_shoulder'),
15:
dict(
name='right_elbow',
id=15,
color=[255, 128, 0],
type='upper',
swap='left_elbow'),
16:
dict(
name='right_wrist',
id=16,
color=[255, 128, 0],
type='upper',
swap='left_wrist')
},
skeleton_info={
0:
dict(link=('root', 'left_hip'), id=0, color=[0, 255, 0]),
1:
dict(link=('left_hip', 'left_knee'), id=1, color=[0, 255, 0]),
2:
dict(link=('left_knee', 'left_foot'), id=2, color=[0, 255, 0]),
3:
dict(link=('root', 'right_hip'), id=3, color=[255, 128, 0]),
4:
dict(link=('right_hip', 'right_knee'), id=4, color=[255, 128, 0]),
5:
dict(link=('right_knee', 'right_foot'), id=5, color=[255, 128, 0]),
6:
dict(link=('root', 'spine'), id=6, color=[51, 153, 255]),
7:
dict(link=('spine', 'thorax'), id=7, color=[51, 153, 255]),
8:
dict(link=('thorax', 'neck_base'), id=8, color=[51, 153, 255]),
9:
dict(link=('neck_base', 'head'), id=9, color=[51, 153, 255]),
10:
dict(link=('thorax', 'left_shoulder'), id=10, color=[0, 255, 0]),
11:
dict(link=('left_shoulder', 'left_elbow'), id=11, color=[0, 255, 0]),
12:
dict(link=('left_elbow', 'left_wrist'), id=12, color=[0, 255, 0]),
13:
dict(link=('thorax', 'right_shoulder'), id=13, color=[255, 128, 0]),
14:
dict(
link=('right_shoulder', 'right_elbow'), id=14, color=[255, 128,
0]),
15:
dict(link=('right_elbow', 'right_wrist'), id=15, color=[255, 128, 0])
},
joint_weights=[1.] * 17,
sigmas=[],
stats_info=dict(bbox_center=(528., 427.), bbox_scale=400.))
configs/_base_/datasets/h3wb.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='h3wb',
paper_info=dict(
author='Yue Zhu, Nermin Samet, David Picard',
title='H3WB: Human3.6M 3D WholeBody Dataset and Benchmark',
container='International Conf. on Computer Vision (ICCV)',
year='2023',
homepage='https://github.com/wholebody3d/wholebody3d',
),
keypoint_info={
0:
dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
1:
dict(
name='left_eye',
id=1,
color=[51, 153, 255],
type='upper',
swap='right_eye'),
2:
dict(
name='right_eye',
id=2,
color=[51, 153, 255],
type='upper',
swap='left_eye'),
3:
dict(
name='left_ear',
id=3,
color=[51, 153, 255],
type='upper',
swap='right_ear'),
4:
dict(
name='right_ear',
id=4,
color=[51, 153, 255],
type='upper',
swap='left_ear'),
5:
dict(
name='left_shoulder',
id=5,
color=[0, 255, 0],
type='upper',
swap='right_shoulder'),
6:
dict(
name='right_shoulder',
id=6,
color=[255, 128, 0],
type='upper',
swap='left_shoulder'),
7:
dict(
name='left_elbow',
id=7,
color=[0, 255, 0],
type='upper',
swap='right_elbow'),
8:
dict(
name='right_elbow',
id=8,
color=[255, 128, 0],
type='upper',
swap='left_elbow'),
9:
dict(
name='left_wrist',
id=9,
color=[0, 255, 0],
type='upper',
swap='right_wrist'),
10:
dict(
name='right_wrist',
id=10,
color=[255, 128, 0],
type='upper',
swap='left_wrist'),
11:
dict(
name='left_hip',
id=11,
color=[0, 255, 0],
type='lower',
swap='right_hip'),
12:
dict(
name='right_hip',
id=12,
color=[255, 128, 0],
type='lower',
swap='left_hip'),
13:
dict(
name='left_knee',
id=13,
color=[0, 255, 0],
type='lower',
swap='right_knee'),
14:
dict(
name='right_knee',
id=14,
color=[255, 128, 0],
type='lower',
swap='left_knee'),
15:
dict(
name='left_ankle',
id=15,
color=[0, 255, 0],
type='lower',
swap='right_ankle'),
16:
dict(
name='right_ankle',
id=16,
color=[255, 128, 0],
type='lower',
swap='left_ankle'),
17:
dict(
name='left_big_toe',
id=17,
color=[255, 128, 0],
type='lower',
swap='right_big_toe'),
18:
dict(
name='left_small_toe',
id=18,
color=[255, 128, 0],
type='lower',
swap='right_small_toe'),
19:
dict(
name='left_heel',
id=19,
color=[255, 128, 0],
type='lower',
swap='right_heel'),
20:
dict(
name='right_big_toe',
id=20,
color=[255, 128, 0],
type='lower',
swap='left_big_toe'),
21:
dict(
name='right_small_toe',
id=21,
color=[255, 128, 0],
type='lower',
swap='left_small_toe'),
22:
dict(
name='right_heel',
id=22,
color=[255, 128, 0],
type='lower',
swap='left_heel'),
23:
dict(
name='face-0',
id=23,
color=[255, 255, 255],
type='',
swap='face-16'),
24:
dict(
name='face-1',
id=24,
color=[255, 255, 255],
type='',
swap='face-15'),
25:
dict(
name='face-2',
id=25,
color=[255, 255, 255],
type='',
swap='face-14'),
26:
dict(
name='face-3',
id=26,
color=[255, 255, 255],
type='',
swap='face-13'),
27:
dict(
name='face-4',
id=27,
color=[255, 255, 255],
type='',
swap='face-12'),
28:
dict(
name='face-5',
id=28,
color=[255, 255, 255],
type='',
swap='face-11'),
29:
dict(
name='face-6',
id=29,
color=[255, 255, 255],
type='',
swap='face-10'),
30:
dict(
name='face-7',
id=30,
color=[255, 255, 255],
type='',
swap='face-9'),
31:
dict(name='face-8', id=31, color=[255, 255, 255], type='', swap=''),
32:
dict(
name='face-9',
id=32,
color=[255, 255, 255],
type='',
swap='face-7'),
33:
dict(
name='face-10',
id=33,
color=[255, 255, 255],
type='',
swap='face-6'),
34:
dict(
name='face-11',
id=34,
color=[255, 255, 255],
type='',
swap='face-5'),
35:
dict(
name='face-12',
id=35,
color=[255, 255, 255],
type='',
swap='face-4'),
36:
dict(
name='face-13',
id=36,
color=[255, 255, 255],
type='',
swap='face-3'),
37:
dict(
name='face-14',
id=37,
color=[255, 255, 255],
type='',
swap='face-2'),
38:
dict(
name='face-15',
id=38,
color=[255, 255, 255],
type='',
swap='face-1'),
39:
dict(
name='face-16',
id=39,
color=[255, 255, 255],
type='',
swap='face-0'),
40:
dict(
name='face-17',
id=40,
color=[255, 255, 255],
type='',
swap='face-26'),
41:
dict(
name='face-18',
id=41,
color=[255, 255, 255],
type='',
swap='face-25'),
42:
dict(
name='face-19',
id=42,
color=[255, 255, 255],
type='',
swap='face-24'),
43:
dict(
name='face-20',
id=43,
color=[255, 255, 255],
type='',
swap='face-23'),
44:
dict(
name='face-21',
id=44,
color=[255, 255, 255],
type='',
swap='face-22'),
45:
dict(
name='face-22',
id=45,
color=[255, 255, 255],
type='',
swap='face-21'),
46:
dict(
name='face-23',
id=46,
color=[255, 255, 255],
type='',
swap='face-20'),
47:
dict(
name='face-24',
id=47,
color=[255, 255, 255],
type='',
swap='face-19'),
48:
dict(
name='face-25',
id=48,
color=[255, 255, 255],
type='',
swap='face-18'),
49:
dict(
name='face-26',
id=49,
color=[255, 255, 255],
type='',
swap='face-17'),
50:
dict(name='face-27', id=50, color=[255, 255, 255], type='', swap=''),
51:
dict(name='face-28', id=51, color=[255, 255, 255], type='', swap=''),
52:
dict(name='face-29', id=52, color=[255, 255, 255], type='', swap=''),
53:
dict(name='face-30', id=53, color=[255, 255, 255], type='', swap=''),
54:
dict(
name='face-31',
id=54,
color=[255, 255, 255],
type='',
swap='face-35'),
55:
dict(
name='face-32',
id=55,
color=[255, 255, 255],
type='',
swap='face-34'),
56:
dict(name='face-33', id=56, color=[255, 255, 255], type='', swap=''),
57:
dict(
name='face-34',
id=57,
color=[255, 255, 255],
type='',
swap='face-32'),
58:
dict(
name='face-35',
id=58,
color=[255, 255, 255],
type='',
swap='face-31'),
59:
dict(
name='face-36',
id=59,
color=[255, 255, 255],
type='',
swap='face-45'),
60:
dict(
name='face-37',
id=60,
color=[255, 255, 255],
type='',
swap='face-44'),
61:
dict(
name='face-38',
id=61,
color=[255, 255, 255],
type='',
swap='face-43'),
62:
dict(
name='face-39',
id=62,
color=[255, 255, 255],
type='',
swap='face-42'),
63:
dict(
name='face-40',
id=63,
color=[255, 255, 255],
type='',
swap='face-47'),
64:
dict(
name='face-41',
id=64,
color=[255, 255, 255],
type='',
swap='face-46'),
65:
dict(
name='face-42',
id=65,
color=[255, 255, 255],
type='',
swap='face-39'),
66:
dict(
name='face-43',
id=66,
color=[255, 255, 255],
type='',
swap='face-38'),
67:
dict(
name='face-44',
id=67,
color=[255, 255, 255],
type='',
swap='face-37'),
68:
dict(
name='face-45',
id=68,
color=[255, 255, 255],
type='',
swap='face-36'),
69:
dict(
name='face-46',
id=69,
color=[255, 255, 255],
type='',
swap='face-41'),
70:
dict(
name='face-47',
id=70,
color=[255, 255, 255],
type='',
swap='face-40'),
71:
dict(
name='face-48',
id=71,
color=[255, 255, 255],
type='',
swap='face-54'),
72:
dict(
name='face-49',
id=72,
color=[255, 255, 255],
type='',
swap='face-53'),
73:
dict(
name='face-50',
id=73,
color=[255, 255, 255],
type='',
swap='face-52'),
74:
dict(name='face-51', id=74, color=[255, 255, 255], type='', swap=''),
75:
dict(
name='face-52',
id=75,
color=[255, 255, 255],
type='',
swap='face-50'),
76:
dict(
name='face-53',
id=76,
color=[255, 255, 255],
type='',
swap='face-49'),
77:
dict(
name='face-54',
id=77,
color=[255, 255, 255],
type='',
swap='face-48'),
78:
dict(
name='face-55',
id=78,
color=[255, 255, 255],
type='',
swap='face-59'),
79:
dict(
name='face-56',
id=79,
color=[255, 255, 255],
type='',
swap='face-58'),
80:
dict(name='face-57', id=80, color=[255, 255, 255], type='', swap=''),
81:
dict(
name='face-58',
id=81,
color=[255, 255, 255],
type='',
swap='face-56'),
82:
dict(
name='face-59',
id=82,
color=[255, 255, 255],
type='',
swap='face-55'),
83:
dict(
name='face-60',
id=83,
color=[255, 255, 255],
type='',
swap='face-64'),
84:
dict(
name='face-61',
id=84,
color=[255, 255, 255],
type='',
swap='face-63'),
85:
dict(name='face-62', id=85, color=[255, 255, 255], type='', swap=''),
86:
dict(
name='face-63',
id=86,
color=[255, 255, 255],
type='',
swap='face-61'),
87:
dict(
name='face-64',
id=87,
color=[255, 255, 255],
type='',
swap='face-60'),
88:
dict(
name='face-65',
id=88,
color=[255, 255, 255],
type='',
swap='face-67'),
89:
dict(name='face-66', id=89, color=[255, 255, 255], type='', swap=''),
90:
dict(
name='face-67',
id=90,
color=[255, 255, 255],
type='',
swap='face-65'),
91:
dict(
name='left_hand_root',
id=91,
color=[255, 255, 255],
type='',
swap='right_hand_root'),
92:
dict(
name='left_thumb1',
id=92,
color=[255, 128, 0],
type='',
swap='right_thumb1'),
93:
dict(
name='left_thumb2',
id=93,
color=[255, 128, 0],
type='',
swap='right_thumb2'),
94:
dict(
name='left_thumb3',
id=94,
color=[255, 128, 0],
type='',
swap='right_thumb3'),
95:
dict(
name='left_thumb4',
id=95,
color=[255, 128, 0],
type='',
swap='right_thumb4'),
96:
dict(
name='left_forefinger1',
id=96,
color=[255, 153, 255],
type='',
swap='right_forefinger1'),
97:
dict(
name='left_forefinger2',
id=97,
color=[255, 153, 255],
type='',
swap='right_forefinger2'),
98:
dict(
name='left_forefinger3',
id=98,
color=[255, 153, 255],
type='',
swap='right_forefinger3'),
99:
dict(
name='left_forefinger4',
id=99,
color=[255, 153, 255],
type='',
swap='right_forefinger4'),
100:
dict(
name='left_middle_finger1',
id=100,
color=[102, 178, 255],
type='',
swap='right_middle_finger1'),
101:
dict(
name='left_middle_finger2',
id=101,
color=[102, 178, 255],
type='',
swap='right_middle_finger2'),
102:
dict(
name='left_middle_finger3',
id=102,
color=[102, 178, 255],
type='',
swap='right_middle_finger3'),
103:
dict(
name='left_middle_finger4',
id=103,
color=[102, 178, 255],
type='',
swap='right_middle_finger4'),
104:
dict(
name='left_ring_finger1',
id=104,
color=[255, 51, 51],
type='',
swap='right_ring_finger1'),
105:
dict(
name='left_ring_finger2',
id=105,
color=[255, 51, 51],
type='',
swap='right_ring_finger2'),
106:
dict(
name='left_ring_finger3',
id=106,
color=[255, 51, 51],
type='',
swap='right_ring_finger3'),
107:
dict(
name='left_ring_finger4',
id=107,
color=[255, 51, 51],
type='',
swap='right_ring_finger4'),
108:
dict(
name='left_pinky_finger1',
id=108,
color=[0, 255, 0],
type='',
swap='right_pinky_finger1'),
109:
dict(
name='left_pinky_finger2',
id=109,
color=[0, 255, 0],
type='',
swap='right_pinky_finger2'),
110:
dict(
name='left_pinky_finger3',
id=110,
color=[0, 255, 0],
type='',
swap='right_pinky_finger3'),
111:
dict(
name='left_pinky_finger4',
id=111,
color=[0, 255, 0],
type='',
swap='right_pinky_finger4'),
112:
dict(
name='right_hand_root',
id=112,
color=[255, 255, 255],
type='',
swap='left_hand_root'),
113:
dict(
name='right_thumb1',
id=113,
color=[255, 128, 0],
type='',
swap='left_thumb1'),
114:
dict(
name='right_thumb2',
id=114,
color=[255, 128, 0],
type='',
swap='left_thumb2'),
115:
dict(
name='right_thumb3',
id=115,
color=[255, 128, 0],
type='',
swap='left_thumb3'),
116:
dict(
name='right_thumb4',
id=116,
color=[255, 128, 0],
type='',
swap='left_thumb4'),
117:
dict(
name='right_forefinger1',
id=117,
color=[255, 153, 255],
type='',
swap='left_forefinger1'),
118:
dict(
name='right_forefinger2',
id=118,
color=[255, 153, 255],
type='',
swap='left_forefinger2'),
119:
dict(
name='right_forefinger3',
id=119,
color=[255, 153, 255],
type='',
swap='left_forefinger3'),
120:
dict(
name='right_forefinger4',
id=120,
color=[255, 153, 255],
type='',
swap='left_forefinger4'),
121:
dict(
name='right_middle_finger1',
id=121,
color=[102, 178, 255],
type='',
swap='left_middle_finger1'),
122:
dict(
name='right_middle_finger2',
id=122,
color=[102, 178, 255],
type='',
swap='left_middle_finger2'),
123:
dict(
name='right_middle_finger3',
id=123,
color=[102, 178, 255],
type='',
swap='left_middle_finger3'),
124:
dict(
name='right_middle_finger4',
id=124,
color=[102, 178, 255],
type='',
swap='left_middle_finger4'),
125:
dict(
name='right_ring_finger1',
id=125,
color=[255, 51, 51],
type='',
swap='left_ring_finger1'),
126:
dict(
name='right_ring_finger2',
id=126,
color=[255, 51, 51],
type='',
swap='left_ring_finger2'),
127:
dict(
name='right_ring_finger3',
id=127,
color=[255, 51, 51],
type='',
swap='left_ring_finger3'),
128:
dict(
name='right_ring_finger4',
id=128,
color=[255, 51, 51],
type='',
swap='left_ring_finger4'),
129:
dict(
name='right_pinky_finger1',
id=129,
color=[0, 255, 0],
type='',
swap='left_pinky_finger1'),
130:
dict(
name='right_pinky_finger2',
id=130,
color=[0, 255, 0],
type='',
swap='left_pinky_finger2'),
131:
dict(
name='right_pinky_finger3',
id=131,
color=[0, 255, 0],
type='',
swap='left_pinky_finger3'),
132:
dict(
name='right_pinky_finger4',
id=132,
color=[0, 255, 0],
type='',
swap='left_pinky_finger4')
},
skeleton_info={
0:
dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
1:
dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
2:
dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
3:
dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
4:
dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
5:
dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
6:
dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
7:
dict(
link=('left_shoulder', 'right_shoulder'),
id=7,
color=[51, 153, 255]),
8:
dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
9:
dict(
link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
10:
dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
11:
dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
12:
dict(link=('left_eye', 'right_eye'), id=12, color=[51, 153, 255]),
13:
dict(link=('nose', 'left_eye'), id=13, color=[51, 153, 255]),
14:
dict(link=('nose', 'right_eye'), id=14, color=[51, 153, 255]),
15:
dict(link=('left_eye', 'left_ear'), id=15, color=[51, 153, 255]),
16:
dict(link=('right_eye', 'right_ear'), id=16, color=[51, 153, 255]),
17:
dict(link=('left_ear', 'left_shoulder'), id=17, color=[51, 153, 255]),
18:
dict(
link=('right_ear', 'right_shoulder'), id=18, color=[51, 153, 255]),
19:
dict(link=('left_ankle', 'left_big_toe'), id=19, color=[0, 255, 0]),
20:
dict(link=('left_ankle', 'left_small_toe'), id=20, color=[0, 255, 0]),
21:
dict(link=('left_ankle', 'left_heel'), id=21, color=[0, 255, 0]),
22:
dict(
link=('right_ankle', 'right_big_toe'), id=22, color=[255, 128, 0]),
23:
dict(
link=('right_ankle', 'right_small_toe'),
id=23,
color=[255, 128, 0]),
24:
dict(link=('right_ankle', 'right_heel'), id=24, color=[255, 128, 0]),
25:
dict(
link=('left_hand_root', 'left_thumb1'), id=25, color=[255, 128,
0]),
26:
dict(link=('left_thumb1', 'left_thumb2'), id=26, color=[255, 128, 0]),
27:
dict(link=('left_thumb2', 'left_thumb3'), id=27, color=[255, 128, 0]),
28:
dict(link=('left_thumb3', 'left_thumb4'), id=28, color=[255, 128, 0]),
29:
dict(
link=('left_hand_root', 'left_forefinger1'),
id=29,
color=[255, 153, 255]),
30:
dict(
link=('left_forefinger1', 'left_forefinger2'),
id=30,
color=[255, 153, 255]),
31:
dict(
link=('left_forefinger2', 'left_forefinger3'),
id=31,
color=[255, 153, 255]),
32:
dict(
link=('left_forefinger3', 'left_forefinger4'),
id=32,
color=[255, 153, 255]),
33:
dict(
link=('left_hand_root', 'left_middle_finger1'),
id=33,
color=[102, 178, 255]),
34:
dict(
link=('left_middle_finger1', 'left_middle_finger2'),
id=34,
color=[102, 178, 255]),
35:
dict(
link=('left_middle_finger2', 'left_middle_finger3'),
id=35,
color=[102, 178, 255]),
36:
dict(
link=('left_middle_finger3', 'left_middle_finger4'),
id=36,
color=[102, 178, 255]),
37:
dict(
link=('left_hand_root', 'left_ring_finger1'),
id=37,
color=[255, 51, 51]),
38:
dict(
link=('left_ring_finger1', 'left_ring_finger2'),
id=38,
color=[255, 51, 51]),
39:
dict(
link=('left_ring_finger2', 'left_ring_finger3'),
id=39,
color=[255, 51, 51]),
40:
dict(
link=('left_ring_finger3', 'left_ring_finger4'),
id=40,
color=[255, 51, 51]),
41:
dict(
link=('left_hand_root', 'left_pinky_finger1'),
id=41,
color=[0, 255, 0]),
42:
dict(
link=('left_pinky_finger1', 'left_pinky_finger2'),
id=42,
color=[0, 255, 0]),
43:
dict(
link=('left_pinky_finger2', 'left_pinky_finger3'),
id=43,
color=[0, 255, 0]),
44:
dict(
link=('left_pinky_finger3', 'left_pinky_finger4'),
id=44,
color=[0, 255, 0]),
45:
dict(
link=('right_hand_root', 'right_thumb1'),
id=45,
color=[255, 128, 0]),
46:
dict(
link=('right_thumb1', 'right_thumb2'), id=46, color=[255, 128, 0]),
47:
dict(
link=('right_thumb2', 'right_thumb3'), id=47, color=[255, 128, 0]),
48:
dict(
link=('right_thumb3', 'right_thumb4'), id=48, color=[255, 128, 0]),
49:
dict(
link=('right_hand_root', 'right_forefinger1'),
id=49,
color=[255, 153, 255]),
50:
dict(
link=('right_forefinger1', 'right_forefinger2'),
id=50,
color=[255, 153, 255]),
51:
dict(
link=('right_forefinger2', 'right_forefinger3'),
id=51,
color=[255, 153, 255]),
52:
dict(
link=('right_forefinger3', 'right_forefinger4'),
id=52,
color=[255, 153, 255]),
53:
dict(
link=('right_hand_root', 'right_middle_finger1'),
id=53,
color=[102, 178, 255]),
54:
dict(
link=('right_middle_finger1', 'right_middle_finger2'),
id=54,
color=[102, 178, 255]),
55:
dict(
link=('right_middle_finger2', 'right_middle_finger3'),
id=55,
color=[102, 178, 255]),
56:
dict(
link=('right_middle_finger3', 'right_middle_finger4'),
id=56,
color=[102, 178, 255]),
57:
dict(
link=('right_hand_root', 'right_ring_finger1'),
id=57,
color=[255, 51, 51]),
58:
dict(
link=('right_ring_finger1', 'right_ring_finger2'),
id=58,
color=[255, 51, 51]),
59:
dict(
link=('right_ring_finger2', 'right_ring_finger3'),
id=59,
color=[255, 51, 51]),
60:
dict(
link=('right_ring_finger3', 'right_ring_finger4'),
id=60,
color=[255, 51, 51]),
61:
dict(
link=('right_hand_root', 'right_pinky_finger1'),
id=61,
color=[0, 255, 0]),
62:
dict(
link=('right_pinky_finger1', 'right_pinky_finger2'),
id=62,
color=[0, 255, 0]),
63:
dict(
link=('right_pinky_finger2', 'right_pinky_finger3'),
id=63,
color=[0, 255, 0]),
64:
dict(
link=('right_pinky_finger3', 'right_pinky_finger4'),
id=64,
color=[0, 255, 0])
},
joint_weights=[1.] * 133,
# 'https://github.com/jin-s13/COCO-WholeBody/blob/master/'
# 'evaluation/myeval_wholebody.py#L175'
sigmas=[
0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089, 0.068, 0.066, 0.066,
0.092, 0.094, 0.094, 0.042, 0.043, 0.044, 0.043, 0.040, 0.035, 0.031,
0.025, 0.020, 0.023, 0.029, 0.032, 0.037, 0.038, 0.043, 0.041, 0.045,
0.013, 0.012, 0.011, 0.011, 0.012, 0.012, 0.011, 0.011, 0.013, 0.015,
0.009, 0.007, 0.007, 0.007, 0.012, 0.009, 0.008, 0.016, 0.010, 0.017,
0.011, 0.009, 0.011, 0.009, 0.007, 0.013, 0.008, 0.011, 0.012, 0.010,
0.034, 0.008, 0.008, 0.009, 0.008, 0.008, 0.007, 0.010, 0.008, 0.009,
0.009, 0.009, 0.007, 0.007, 0.008, 0.011, 0.008, 0.008, 0.008, 0.01,
0.008, 0.029, 0.022, 0.035, 0.037, 0.047, 0.026, 0.025, 0.024, 0.035,
0.018, 0.024, 0.022, 0.026, 0.017, 0.021, 0.021, 0.032, 0.02, 0.019,
0.022, 0.031, 0.029, 0.022, 0.035, 0.037, 0.047, 0.026, 0.025, 0.024,
0.035, 0.018, 0.024, 0.022, 0.026, 0.017, 0.021, 0.021, 0.032, 0.02,
0.019, 0.022, 0.031
])
configs/_base_/datasets/halpe.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='halpe',
paper_info=dict(
author='Li, Yong-Lu and Xu, Liang and Liu, Xinpeng and Huang, Xijie'
' and Xu, Yue and Wang, Shiyi and Fang, Hao-Shu'
' and Ma, Ze and Chen, Mingyang and Lu, Cewu',
title='PaStaNet: Toward Human Activity Knowledge Engine',
container='CVPR',
year='2020',
homepage='https://github.com/Fang-Haoshu/Halpe-FullBody/',
),
keypoint_info={
0:
dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
1:
dict(
name='left_eye',
id=1,
color=[51, 153, 255],
type='upper',
swap='right_eye'),
2:
dict(
name='right_eye',
id=2,
color=[51, 153, 255],
type='upper',
swap='left_eye'),
3:
dict(
name='left_ear',
id=3,
color=[51, 153, 255],
type='upper',
swap='right_ear'),
4:
dict(
name='right_ear',
id=4,
color=[51, 153, 255],
type='upper',
swap='left_ear'),
5:
dict(
name='left_shoulder',
id=5,
color=[0, 255, 0],
type='upper',
swap='right_shoulder'),
6:
dict(
name='right_shoulder',
id=6,
color=[255, 128, 0],
type='upper',
swap='left_shoulder'),
7:
dict(
name='left_elbow',
id=7,
color=[0, 255, 0],
type='upper',
swap='right_elbow'),
8:
dict(
name='right_elbow',
id=8,
color=[255, 128, 0],
type='upper',
swap='left_elbow'),
9:
dict(
name='left_wrist',
id=9,
color=[0, 255, 0],
type='upper',
swap='right_wrist'),
10:
dict(
name='right_wrist',
id=10,
color=[255, 128, 0],
type='upper',
swap='left_wrist'),
11:
dict(
name='left_hip',
id=11,
color=[0, 255, 0],
type='lower',
swap='right_hip'),
12:
dict(
name='right_hip',
id=12,
color=[255, 128, 0],
type='lower',
swap='left_hip'),
13:
dict(
name='left_knee',
id=13,
color=[0, 255, 0],
type='lower',
swap='right_knee'),
14:
dict(
name='right_knee',
id=14,
color=[255, 128, 0],
type='lower',
swap='left_knee'),
15:
dict(
name='left_ankle',
id=15,
color=[0, 255, 0],
type='lower',
swap='right_ankle'),
16:
dict(
name='right_ankle',
id=16,
color=[255, 128, 0],
type='lower',
swap='left_ankle'),
17:
dict(name='head', id=17, color=[255, 128, 0], type='upper', swap=''),
18:
dict(name='neck', id=18, color=[255, 128, 0], type='upper', swap=''),
19:
dict(name='hip', id=19, color=[255, 128, 0], type='lower', swap=''),
20:
dict(
name='left_big_toe',
id=20,
color=[255, 128, 0],
type='lower',
swap='right_big_toe'),
21:
dict(
name='right_big_toe',
id=21,
color=[255, 128, 0],
type='lower',
swap='left_big_toe'),
22:
dict(
name='left_small_toe',
id=22,
color=[255, 128, 0],
type='lower',
swap='right_small_toe'),
23:
dict(
name='right_small_toe',
id=23,
color=[255, 128, 0],
type='lower',
swap='left_small_toe'),
24:
dict(
name='left_heel',
id=24,
color=[255, 128, 0],
type='lower',
swap='right_heel'),
25:
dict(
name='right_heel',
id=25,
color=[255, 128, 0],
type='lower',
swap='left_heel'),
26:
dict(
name='face-0',
id=26,
color=[255, 255, 255],
type='',
swap='face-16'),
27:
dict(
name='face-1',
id=27,
color=[255, 255, 255],
type='',
swap='face-15'),
28:
dict(
name='face-2',
id=28,
color=[255, 255, 255],
type='',
swap='face-14'),
29:
dict(
name='face-3',
id=29,
color=[255, 255, 255],
type='',
swap='face-13'),
30:
dict(
name='face-4',
id=30,
color=[255, 255, 255],
type='',
swap='face-12'),
31:
dict(
name='face-5',
id=31,
color=[255, 255, 255],
type='',
swap='face-11'),
32:
dict(
name='face-6',
id=32,
color=[255, 255, 255],
type='',
swap='face-10'),
33:
dict(
name='face-7',
id=33,
color=[255, 255, 255],
type='',
swap='face-9'),
34:
dict(name='face-8', id=34, color=[255, 255, 255], type='', swap=''),
35:
dict(
name='face-9',
id=35,
color=[255, 255, 255],
type='',
swap='face-7'),
36:
dict(
name='face-10',
id=36,
color=[255, 255, 255],
type='',
swap='face-6'),
37:
dict(
name='face-11',
id=37,
color=[255, 255, 255],
type='',
swap='face-5'),
38:
dict(
name='face-12',
id=38,
color=[255, 255, 255],
type='',
swap='face-4'),
39:
dict(
name='face-13',
id=39,
color=[255, 255, 255],
type='',
swap='face-3'),
40:
dict(
name='face-14',
id=40,
color=[255, 255, 255],
type='',
swap='face-2'),
41:
dict(
name='face-15',
id=41,
color=[255, 255, 255],
type='',
swap='face-1'),
42:
dict(
name='face-16',
id=42,
color=[255, 255, 255],
type='',
swap='face-0'),
43:
dict(
name='face-17',
id=43,
color=[255, 255, 255],
type='',
swap='face-26'),
44:
dict(
name='face-18',
id=44,
color=[255, 255, 255],
type='',
swap='face-25'),
45:
dict(
name='face-19',
id=45,
color=[255, 255, 255],
type='',
swap='face-24'),
46:
dict(
name='face-20',
id=46,
color=[255, 255, 255],
type='',
swap='face-23'),
47:
dict(
name='face-21',
id=47,
color=[255, 255, 255],
type='',
swap='face-22'),
48:
dict(
name='face-22',
id=48,
color=[255, 255, 255],
type='',
swap='face-21'),
49:
dict(
name='face-23',
id=49,
color=[255, 255, 255],
type='',
swap='face-20'),
50:
dict(
name='face-24',
id=50,
color=[255, 255, 255],
type='',
swap='face-19'),
51:
dict(
name='face-25',
id=51,
color=[255, 255, 255],
type='',
swap='face-18'),
52:
dict(
name='face-26',
id=52,
color=[255, 255, 255],
type='',
swap='face-17'),
53:
dict(name='face-27', id=53, color=[255, 255, 255], type='', swap=''),
54:
dict(name='face-28', id=54, color=[255, 255, 255], type='', swap=''),
55:
dict(name='face-29', id=55, color=[255, 255, 255], type='', swap=''),
56:
dict(name='face-30', id=56, color=[255, 255, 255], type='', swap=''),
57:
dict(
name='face-31',
id=57,
color=[255, 255, 255],
type='',
swap='face-35'),
58:
dict(
name='face-32',
id=58,
color=[255, 255, 255],
type='',
swap='face-34'),
59:
dict(name='face-33', id=59, color=[255, 255, 255], type='', swap=''),
60:
dict(
name='face-34',
id=60,
color=[255, 255, 255],
type='',
swap='face-32'),
61:
dict(
name='face-35',
id=61,
color=[255, 255, 255],
type='',
swap='face-31'),
62:
dict(
name='face-36',
id=62,
color=[255, 255, 255],
type='',
swap='face-45'),
63:
dict(
name='face-37',
id=63,
color=[255, 255, 255],
type='',
swap='face-44'),
64:
dict(
name='face-38',
id=64,
color=[255, 255, 255],
type='',
swap='face-43'),
65:
dict(
name='face-39',
id=65,
color=[255, 255, 255],
type='',
swap='face-42'),
66:
dict(
name='face-40',
id=66,
color=[255, 255, 255],
type='',
swap='face-47'),
67:
dict(
name='face-41',
id=67,
color=[255, 255, 255],
type='',
swap='face-46'),
68:
dict(
name='face-42',
id=68,
color=[255, 255, 255],
type='',
swap='face-39'),
69:
dict(
name='face-43',
id=69,
color=[255, 255, 255],
type='',
swap='face-38'),
70:
dict(
name='face-44',
id=70,
color=[255, 255, 255],
type='',
swap='face-37'),
71:
dict(
name='face-45',
id=71,
color=[255, 255, 255],
type='',
swap='face-36'),
72:
dict(
name='face-46',
id=72,
color=[255, 255, 255],
type='',
swap='face-41'),
73:
dict(
name='face-47',
id=73,
color=[255, 255, 255],
type='',
swap='face-40'),
74:
dict(
name='face-48',
id=74,
color=[255, 255, 255],
type='',
swap='face-54'),
75:
dict(
name='face-49',
id=75,
color=[255, 255, 255],
type='',
swap='face-53'),
76:
dict(
name='face-50',
id=76,
color=[255, 255, 255],
type='',
swap='face-52'),
77:
dict(name='face-51', id=77, color=[255, 255, 255], type='', swap=''),
78:
dict(
name='face-52',
id=78,
color=[255, 255, 255],
type='',
swap='face-50'),
79:
dict(
name='face-53',
id=79,
color=[255, 255, 255],
type='',
swap='face-49'),
80:
dict(
name='face-54',
id=80,
color=[255, 255, 255],
type='',
swap='face-48'),
81:
dict(
name='face-55',
id=81,
color=[255, 255, 255],
type='',
swap='face-59'),
82:
dict(
name='face-56',
id=82,
color=[255, 255, 255],
type='',
swap='face-58'),
83:
dict(name='face-57', id=83, color=[255, 255, 255], type='', swap=''),
84:
dict(
name='face-58',
id=84,
color=[255, 255, 255],
type='',
swap='face-56'),
85:
dict(
name='face-59',
id=85,
color=[255, 255, 255],
type='',
swap='face-55'),
86:
dict(
name='face-60',
id=86,
color=[255, 255, 255],
type='',
swap='face-64'),
87:
dict(
name='face-61',
id=87,
color=[255, 255, 255],
type='',
swap='face-63'),
88:
dict(name='face-62', id=88, color=[255, 255, 255], type='', swap=''),
89:
dict(
name='face-63',
id=89,
color=[255, 255, 255],
type='',
swap='face-61'),
90:
dict(
name='face-64',
id=90,
color=[255, 255, 255],
type='',
swap='face-60'),
91:
dict(
name='face-65',
id=91,
color=[255, 255, 255],
type='',
swap='face-67'),
92:
dict(name='face-66', id=92, color=[255, 255, 255], type='', swap=''),
93:
dict(
name='face-67',
id=93,
color=[255, 255, 255],
type='',
swap='face-65'),
94:
dict(
name='left_hand_root',
id=94,
color=[255, 255, 255],
type='',
swap='right_hand_root'),
95:
dict(
name='left_thumb1',
id=95,
color=[255, 128, 0],
type='',
swap='right_thumb1'),
96:
dict(
name='left_thumb2',
id=96,
color=[255, 128, 0],
type='',
swap='right_thumb2'),
97:
dict(
name='left_thumb3',
id=97,
color=[255, 128, 0],
type='',
swap='right_thumb3'),
98:
dict(
name='left_thumb4',
id=98,
color=[255, 128, 0],
type='',
swap='right_thumb4'),
99:
dict(
name='left_forefinger1',
id=99,
color=[255, 153, 255],
type='',
swap='right_forefinger1'),
100:
dict(
name='left_forefinger2',
id=100,
color=[255, 153, 255],
type='',
swap='right_forefinger2'),
101:
dict(
name='left_forefinger3',
id=101,
color=[255, 153, 255],
type='',
swap='right_forefinger3'),
102:
dict(
name='left_forefinger4',
id=102,
color=[255, 153, 255],
type='',
swap='right_forefinger4'),
103:
dict(
name='left_middle_finger1',
id=103,
color=[102, 178, 255],
type='',
swap='right_middle_finger1'),
104:
dict(
name='left_middle_finger2',
id=104,
color=[102, 178, 255],
type='',
swap='right_middle_finger2'),
105:
dict(
name='left_middle_finger3',
id=105,
color=[102, 178, 255],
type='',
swap='right_middle_finger3'),
106:
dict(
name='left_middle_finger4',
id=106,
color=[102, 178, 255],
type='',
swap='right_middle_finger4'),
107:
dict(
name='left_ring_finger1',
id=107,
color=[255, 51, 51],
type='',
swap='right_ring_finger1'),
108:
dict(
name='left_ring_finger2',
id=108,
color=[255, 51, 51],
type='',
swap='right_ring_finger2'),
109:
dict(
name='left_ring_finger3',
id=109,
color=[255, 51, 51],
type='',
swap='right_ring_finger3'),
110:
dict(
name='left_ring_finger4',
id=110,
color=[255, 51, 51],
type='',
swap='right_ring_finger4'),
111:
dict(
name='left_pinky_finger1',
id=111,
color=[0, 255, 0],
type='',
swap='right_pinky_finger1'),
112:
dict(
name='left_pinky_finger2',
id=112,
color=[0, 255, 0],
type='',
swap='right_pinky_finger2'),
113:
dict(
name='left_pinky_finger3',
id=113,
color=[0, 255, 0],
type='',
swap='right_pinky_finger3'),
114:
dict(
name='left_pinky_finger4',
id=114,
color=[0, 255, 0],
type='',
swap='right_pinky_finger4'),
115:
dict(
name='right_hand_root',
id=115,
color=[255, 255, 255],
type='',
swap='left_hand_root'),
116:
dict(
name='right_thumb1',
id=116,
color=[255, 128, 0],
type='',
swap='left_thumb1'),
117:
dict(
name='right_thumb2',
id=117,
color=[255, 128, 0],
type='',
swap='left_thumb2'),
118:
dict(
name='right_thumb3',
id=118,
color=[255, 128, 0],
type='',
swap='left_thumb3'),
119:
dict(
name='right_thumb4',
id=119,
color=[255, 128, 0],
type='',
swap='left_thumb4'),
120:
dict(
name='right_forefinger1',
id=120,
color=[255, 153, 255],
type='',
swap='left_forefinger1'),
121:
dict(
name='right_forefinger2',
id=121,
color=[255, 153, 255],
type='',
swap='left_forefinger2'),
122:
dict(
name='right_forefinger3',
id=122,
color=[255, 153, 255],
type='',
swap='left_forefinger3'),
123:
dict(
name='right_forefinger4',
id=123,
color=[255, 153, 255],
type='',
swap='left_forefinger4'),
124:
dict(
name='right_middle_finger1',
id=124,
color=[102, 178, 255],
type='',
swap='left_middle_finger1'),
125:
dict(
name='right_middle_finger2',
id=125,
color=[102, 178, 255],
type='',
swap='left_middle_finger2'),
126:
dict(
name='right_middle_finger3',
id=126,
color=[102, 178, 255],
type='',
swap='left_middle_finger3'),
127:
dict(
name='right_middle_finger4',
id=127,
color=[102, 178, 255],
type='',
swap='left_middle_finger4'),
128:
dict(
name='right_ring_finger1',
id=128,
color=[255, 51, 51],
type='',
swap='left_ring_finger1'),
129:
dict(
name='right_ring_finger2',
id=129,
color=[255, 51, 51],
type='',
swap='left_ring_finger2'),
130:
dict(
name='right_ring_finger3',
id=130,
color=[255, 51, 51],
type='',
swap='left_ring_finger3'),
131:
dict(
name='right_ring_finger4',
id=131,
color=[255, 51, 51],
type='',
swap='left_ring_finger4'),
132:
dict(
name='right_pinky_finger1',
id=132,
color=[0, 255, 0],
type='',
swap='left_pinky_finger1'),
133:
dict(
name='right_pinky_finger2',
id=133,
color=[0, 255, 0],
type='',
swap='left_pinky_finger2'),
134:
dict(
name='right_pinky_finger3',
id=134,
color=[0, 255, 0],
type='',
swap='left_pinky_finger3'),
135:
dict(
name='right_pinky_finger4',
id=135,
color=[0, 255, 0],
type='',
swap='left_pinky_finger4')
},
skeleton_info={
0:
dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
1:
dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
2:
dict(link=('left_hip', 'hip'), id=2, color=[0, 255, 0]),
3:
dict(link=('right_ankle', 'right_knee'), id=3, color=[255, 128, 0]),
4:
dict(link=('right_knee', 'right_hip'), id=4, color=[255, 128, 0]),
5:
dict(link=('right_hip', 'hip'), id=5, color=[255, 128, 0]),
6:
dict(link=('head', 'neck'), id=6, color=[51, 153, 255]),
7:
dict(link=('neck', 'hip'), id=7, color=[51, 153, 255]),
8:
dict(link=('neck', 'left_shoulder'), id=8, color=[0, 255, 0]),
9:
dict(link=('left_shoulder', 'left_elbow'), id=9, color=[0, 255, 0]),
10:
dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
11:
dict(link=('neck', 'right_shoulder'), id=11, color=[255, 128, 0]),
12:
dict(
link=('right_shoulder', 'right_elbow'), id=12, color=[255, 128,
0]),
13:
dict(link=('right_elbow', 'right_wrist'), id=13, color=[255, 128, 0]),
14:
dict(link=('left_eye', 'right_eye'), id=14, color=[51, 153, 255]),
15:
dict(link=('nose', 'left_eye'), id=15, color=[51, 153, 255]),
16:
dict(link=('nose', 'right_eye'), id=16, color=[51, 153, 255]),
17:
dict(link=('left_eye', 'left_ear'), id=17, color=[51, 153, 255]),
18:
dict(link=('right_eye', 'right_ear'), id=18, color=[51, 153, 255]),
19:
dict(link=('left_ear', 'left_shoulder'), id=19, color=[51, 153, 255]),
20:
dict(
link=('right_ear', 'right_shoulder'), id=20, color=[51, 153, 255]),
21:
dict(link=('left_ankle', 'left_big_toe'), id=21, color=[0, 255, 0]),
22:
dict(link=('left_ankle', 'left_small_toe'), id=22, color=[0, 255, 0]),
23:
dict(link=('left_ankle', 'left_heel'), id=23, color=[0, 255, 0]),
24:
dict(
link=('right_ankle', 'right_big_toe'), id=24, color=[255, 128, 0]),
25:
dict(
link=('right_ankle', 'right_small_toe'),
id=25,
color=[255, 128, 0]),
26:
dict(link=('right_ankle', 'right_heel'), id=26, color=[255, 128, 0]),
27:
dict(link=('left_wrist', 'left_thumb1'), id=27, color=[255, 128, 0]),
28:
dict(link=('left_thumb1', 'left_thumb2'), id=28, color=[255, 128, 0]),
29:
dict(link=('left_thumb2', 'left_thumb3'), id=29, color=[255, 128, 0]),
30:
dict(link=('left_thumb3', 'left_thumb4'), id=30, color=[255, 128, 0]),
31:
dict(
link=('left_wrist', 'left_forefinger1'),
id=31,
color=[255, 153, 255]),
32:
dict(
link=('left_forefinger1', 'left_forefinger2'),
id=32,
color=[255, 153, 255]),
33:
dict(
link=('left_forefinger2', 'left_forefinger3'),
id=33,
color=[255, 153, 255]),
34:
dict(
link=('left_forefinger3', 'left_forefinger4'),
id=34,
color=[255, 153, 255]),
35:
dict(
link=('left_wrist', 'left_middle_finger1'),
id=35,
color=[102, 178, 255]),
36:
dict(
link=('left_middle_finger1', 'left_middle_finger2'),
id=36,
color=[102, 178, 255]),
37:
dict(
link=('left_middle_finger2', 'left_middle_finger3'),
id=37,
color=[102, 178, 255]),
38:
dict(
link=('left_middle_finger3', 'left_middle_finger4'),
id=38,
color=[102, 178, 255]),
39:
dict(
link=('left_wrist', 'left_ring_finger1'),
id=39,
color=[255, 51, 51]),
40:
dict(
link=('left_ring_finger1', 'left_ring_finger2'),
id=40,
color=[255, 51, 51]),
41:
dict(
link=('left_ring_finger2', 'left_ring_finger3'),
id=41,
color=[255, 51, 51]),
42:
dict(
link=('left_ring_finger3', 'left_ring_finger4'),
id=42,
color=[255, 51, 51]),
43:
dict(
link=('left_wrist', 'left_pinky_finger1'),
id=43,
color=[0, 255, 0]),
44:
dict(
link=('left_pinky_finger1', 'left_pinky_finger2'),
id=44,
color=[0, 255, 0]),
45:
dict(
link=('left_pinky_finger2', 'left_pinky_finger3'),
id=45,
color=[0, 255, 0]),
46:
dict(
link=('left_pinky_finger3', 'left_pinky_finger4'),
id=46,
color=[0, 255, 0]),
47:
dict(link=('right_wrist', 'right_thumb1'), id=47, color=[255, 128, 0]),
48:
dict(
link=('right_thumb1', 'right_thumb2'), id=48, color=[255, 128, 0]),
49:
dict(
link=('right_thumb2', 'right_thumb3'), id=49, color=[255, 128, 0]),
50:
dict(
link=('right_thumb3', 'right_thumb4'), id=50, color=[255, 128, 0]),
51:
dict(
link=('right_wrist', 'right_forefinger1'),
id=51,
color=[255, 153, 255]),
52:
dict(
link=('right_forefinger1', 'right_forefinger2'),
id=52,
color=[255, 153, 255]),
53:
dict(
link=('right_forefinger2', 'right_forefinger3'),
id=53,
color=[255, 153, 255]),
54:
dict(
link=('right_forefinger3', 'right_forefinger4'),
id=54,
color=[255, 153, 255]),
55:
dict(
link=('right_wrist', 'right_middle_finger1'),
id=55,
color=[102, 178, 255]),
56:
dict(
link=('right_middle_finger1', 'right_middle_finger2'),
id=56,
color=[102, 178, 255]),
57:
dict(
link=('right_middle_finger2', 'right_middle_finger3'),
id=57,
color=[102, 178, 255]),
58:
dict(
link=('right_middle_finger3', 'right_middle_finger4'),
id=58,
color=[102, 178, 255]),
59:
dict(
link=('right_wrist', 'right_ring_finger1'),
id=59,
color=[255, 51, 51]),
60:
dict(
link=('right_ring_finger1', 'right_ring_finger2'),
id=60,
color=[255, 51, 51]),
61:
dict(
link=('right_ring_finger2', 'right_ring_finger3'),
id=61,
color=[255, 51, 51]),
62:
dict(
link=('right_ring_finger3', 'right_ring_finger4'),
id=62,
color=[255, 51, 51]),
63:
dict(
link=('right_wrist', 'right_pinky_finger1'),
id=63,
color=[0, 255, 0]),
64:
dict(
link=('right_pinky_finger1', 'right_pinky_finger2'),
id=64,
color=[0, 255, 0]),
65:
dict(
link=('right_pinky_finger2', 'right_pinky_finger3'),
id=65,
color=[0, 255, 0]),
66:
dict(
link=('right_pinky_finger3', 'right_pinky_finger4'),
id=66,
color=[0, 255, 0])
},
joint_weights=[1.] * 136,
# 'https://github.com/Fang-Haoshu/Halpe-FullBody/blob/master/'
# 'HalpeCOCOAPI/PythonAPI/halpecocotools/cocoeval.py#L245'
sigmas=[
0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089, 0.08, 0.08, 0.08,
0.089, 0.089, 0.089, 0.089, 0.089, 0.089, 0.015, 0.015, 0.015, 0.015,
0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
0.015, 0.015, 0.015, 0.015, 0.015, 0.015
])
configs/_base_/datasets/halpe26.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='halpe26',
paper_info=dict(
author='Li, Yong-Lu and Xu, Liang and Liu, Xinpeng and Huang, Xijie'
' and Xu, Yue and Wang, Shiyi and Fang, Hao-Shu'
' and Ma, Ze and Chen, Mingyang and Lu, Cewu',
title='PaStaNet: Toward Human Activity Knowledge Engine',
container='CVPR',
year='2020',
homepage='https://github.com/Fang-Haoshu/Halpe-FullBody/',
),
keypoint_info={
0:
dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
1:
dict(
name='left_eye',
id=1,
color=[51, 153, 255],
type='upper',
swap='right_eye'),
2:
dict(
name='right_eye',
id=2,
color=[51, 153, 255],
type='upper',
swap='left_eye'),
3:
dict(
name='left_ear',
id=3,
color=[51, 153, 255],
type='upper',
swap='right_ear'),
4:
dict(
name='right_ear',
id=4,
color=[51, 153, 255],
type='upper',
swap='left_ear'),
5:
dict(
name='left_shoulder',
id=5,
color=[0, 255, 0],
type='upper',
swap='right_shoulder'),
6:
dict(
name='right_shoulder',
id=6,
color=[255, 128, 0],
type='upper',
swap='left_shoulder'),
7:
dict(
name='left_elbow',
id=7,
color=[0, 255, 0],
type='upper',
swap='right_elbow'),
8:
dict(
name='right_elbow',
id=8,
color=[255, 128, 0],
type='upper',
swap='left_elbow'),
9:
dict(
name='left_wrist',
id=9,
color=[0, 255, 0],
type='upper',
swap='right_wrist'),
10:
dict(
name='right_wrist',
id=10,
color=[255, 128, 0],
type='upper',
swap='left_wrist'),
11:
dict(
name='left_hip',
id=11,
color=[0, 255, 0],
type='lower',
swap='right_hip'),
12:
dict(
name='right_hip',
id=12,
color=[255, 128, 0],
type='lower',
swap='left_hip'),
13:
dict(
name='left_knee',
id=13,
color=[0, 255, 0],
type='lower',
swap='right_knee'),
14:
dict(
name='right_knee',
id=14,
color=[255, 128, 0],
type='lower',
swap='left_knee'),
15:
dict(
name='left_ankle',
id=15,
color=[0, 255, 0],
type='lower',
swap='right_ankle'),
16:
dict(
name='right_ankle',
id=16,
color=[255, 128, 0],
type='lower',
swap='left_ankle'),
17:
dict(name='head', id=17, color=[255, 128, 0], type='upper', swap=''),
18:
dict(name='neck', id=18, color=[255, 128, 0], type='upper', swap=''),
19:
dict(name='hip', id=19, color=[255, 128, 0], type='lower', swap=''),
20:
dict(
name='left_big_toe',
id=20,
color=[255, 128, 0],
type='lower',
swap='right_big_toe'),
21:
dict(
name='right_big_toe',
id=21,
color=[255, 128, 0],
type='lower',
swap='left_big_toe'),
22:
dict(
name='left_small_toe',
id=22,
color=[255, 128, 0],
type='lower',
swap='right_small_toe'),
23:
dict(
name='right_small_toe',
id=23,
color=[255, 128, 0],
type='lower',
swap='left_small_toe'),
24:
dict(
name='left_heel',
id=24,
color=[255, 128, 0],
type='lower',
swap='right_heel'),
25:
dict(
name='right_heel',
id=25,
color=[255, 128, 0],
type='lower',
swap='left_heel')
},
skeleton_info={
0:
dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
1:
dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
2:
dict(link=('left_hip', 'hip'), id=2, color=[0, 255, 0]),
3:
dict(link=('right_ankle', 'right_knee'), id=3, color=[255, 128, 0]),
4:
dict(link=('right_knee', 'right_hip'), id=4, color=[255, 128, 0]),
5:
dict(link=('right_hip', 'hip'), id=5, color=[255, 128, 0]),
6:
dict(link=('head', 'neck'), id=6, color=[51, 153, 255]),
7:
dict(link=('neck', 'hip'), id=7, color=[51, 153, 255]),
8:
dict(link=('neck', 'left_shoulder'), id=8, color=[0, 255, 0]),
9:
dict(link=('left_shoulder', 'left_elbow'), id=9, color=[0, 255, 0]),
10:
dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
11:
dict(link=('neck', 'right_shoulder'), id=11, color=[255, 128, 0]),
12:
dict(
link=('right_shoulder', 'right_elbow'), id=12, color=[255, 128,
0]),
13:
dict(link=('right_elbow', 'right_wrist'), id=13, color=[255, 128, 0]),
14:
dict(link=('left_eye', 'right_eye'), id=14, color=[51, 153, 255]),
15:
dict(link=('nose', 'left_eye'), id=15, color=[51, 153, 255]),
16:
dict(link=('nose', 'right_eye'), id=16, color=[51, 153, 255]),
17:
dict(link=('left_eye', 'left_ear'), id=17, color=[51, 153, 255]),
18:
dict(link=('right_eye', 'right_ear'), id=18, color=[51, 153, 255]),
19:
dict(link=('left_ear', 'left_shoulder'), id=19, color=[51, 153, 255]),
20:
dict(
link=('right_ear', 'right_shoulder'), id=20, color=[51, 153, 255]),
21:
dict(link=('left_ankle', 'left_big_toe'), id=21, color=[0, 255, 0]),
22:
dict(link=('left_ankle', 'left_small_toe'), id=22, color=[0, 255, 0]),
23:
dict(link=('left_ankle', 'left_heel'), id=23, color=[0, 255, 0]),
24:
dict(
link=('right_ankle', 'right_big_toe'), id=24, color=[255, 128, 0]),
25:
dict(
link=('right_ankle', 'right_small_toe'),
id=25,
color=[255, 128, 0]),
26:
dict(link=('right_ankle', 'right_heel'), id=26, color=[255, 128, 0]),
},
# the joint_weights is modified by MMPose Team
joint_weights=[
1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.5, 1.5, 1., 1., 1.2, 1.2, 1.5,
1.5
] + [1., 1., 1.2] + [1.5] * 6,
# 'https://github.com/Fang-Haoshu/Halpe-FullBody/blob/master/'
# 'HalpeCOCOAPI/PythonAPI/halpecocotools/cocoeval.py#L245'
sigmas=[
0.026,
0.025,
0.025,
0.035,
0.035,
0.079,
0.079,
0.072,
0.072,
0.062,
0.062,
0.107,
0.107,
0.087,
0.087,
0.089,
0.089,
0.026,
0.026,
0.066,
0.079,
0.079,
0.079,
0.079,
0.079,
0.079,
])
configs/_base_/datasets/horse10.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='horse10',
paper_info=dict(
author='Mathis, Alexander and Biasi, Thomas and '
'Schneider, Steffen and '
'Yuksekgonul, Mert and Rogers, Byron and '
'Bethge, Matthias and '
'Mathis, Mackenzie W',
title='Pretraining boosts out-of-domain robustness '
'for pose estimation',
container='Proceedings of the IEEE/CVF Winter Conference on '
'Applications of Computer Vision',
year='2021',
homepage='http://www.mackenziemathislab.org/horse10',
),
keypoint_info={
0:
dict(name='Nose', id=0, color=[255, 153, 255], type='upper', swap=''),
1:
dict(name='Eye', id=1, color=[255, 153, 255], type='upper', swap=''),
2:
dict(
name='Nearknee',
id=2,
color=[255, 102, 255],
type='upper',
swap=''),
3:
dict(
name='Nearfrontfetlock',
id=3,
color=[255, 102, 255],
type='upper',
swap=''),
4:
dict(
name='Nearfrontfoot',
id=4,
color=[255, 102, 255],
type='upper',
swap=''),
5:
dict(
name='Offknee', id=5, color=[255, 102, 255], type='upper',
swap=''),
6:
dict(
name='Offfrontfetlock',
id=6,
color=[255, 102, 255],
type='upper',
swap=''),
7:
dict(
name='Offfrontfoot',
id=7,
color=[255, 102, 255],
type='upper',
swap=''),
8:
dict(
name='Shoulder',
id=8,
color=[255, 153, 255],
type='upper',
swap=''),
9:
dict(
name='Midshoulder',
id=9,
color=[255, 153, 255],
type='upper',
swap=''),
10:
dict(
name='Elbow', id=10, color=[255, 153, 255], type='upper', swap=''),
11:
dict(
name='Girth', id=11, color=[255, 153, 255], type='upper', swap=''),
12:
dict(
name='Wither', id=12, color=[255, 153, 255], type='upper',
swap=''),
13:
dict(
name='Nearhindhock',
id=13,
color=[255, 51, 255],
type='lower',
swap=''),
14:
dict(
name='Nearhindfetlock',
id=14,
color=[255, 51, 255],
type='lower',
swap=''),
15:
dict(
name='Nearhindfoot',
id=15,
color=[255, 51, 255],
type='lower',
swap=''),
16:
dict(name='Hip', id=16, color=[255, 153, 255], type='lower', swap=''),
17:
dict(
name='Stifle', id=17, color=[255, 153, 255], type='lower',
swap=''),
18:
dict(
name='Offhindhock',
id=18,
color=[255, 51, 255],
type='lower',
swap=''),
19:
dict(
name='Offhindfetlock',
id=19,
color=[255, 51, 255],
type='lower',
swap=''),
20:
dict(
name='Offhindfoot',
id=20,
color=[255, 51, 255],
type='lower',
swap=''),
21:
dict(
name='Ischium',
id=21,
color=[255, 153, 255],
type='lower',
swap='')
},
skeleton_info={
0:
dict(link=('Nose', 'Eye'), id=0, color=[255, 153, 255]),
1:
dict(link=('Eye', 'Wither'), id=1, color=[255, 153, 255]),
2:
dict(link=('Wither', 'Hip'), id=2, color=[255, 153, 255]),
3:
dict(link=('Hip', 'Ischium'), id=3, color=[255, 153, 255]),
4:
dict(link=('Ischium', 'Stifle'), id=4, color=[255, 153, 255]),
5:
dict(link=('Stifle', 'Girth'), id=5, color=[255, 153, 255]),
6:
dict(link=('Girth', 'Elbow'), id=6, color=[255, 153, 255]),
7:
dict(link=('Elbow', 'Shoulder'), id=7, color=[255, 153, 255]),
8:
dict(link=('Shoulder', 'Midshoulder'), id=8, color=[255, 153, 255]),
9:
dict(link=('Midshoulder', 'Wither'), id=9, color=[255, 153, 255]),
10:
dict(
link=('Nearknee', 'Nearfrontfetlock'),
id=10,
color=[255, 102, 255]),
11:
dict(
link=('Nearfrontfetlock', 'Nearfrontfoot'),
id=11,
color=[255, 102, 255]),
12:
dict(
link=('Offknee', 'Offfrontfetlock'), id=12, color=[255, 102, 255]),
13:
dict(
link=('Offfrontfetlock', 'Offfrontfoot'),
id=13,
color=[255, 102, 255]),
14:
dict(
link=('Nearhindhock', 'Nearhindfetlock'),
id=14,
color=[255, 51, 255]),
15:
dict(
link=('Nearhindfetlock', 'Nearhindfoot'),
id=15,
color=[255, 51, 255]),
16:
dict(
link=('Offhindhock', 'Offhindfetlock'),
id=16,
color=[255, 51, 255]),
17:
dict(
link=('Offhindfetlock', 'Offhindfoot'),
id=17,
color=[255, 51, 255])
},
joint_weights=[1.] * 22,
sigmas=[])
configs/_base_/datasets/humanart.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='Human-Art',
paper_info=dict(
author='Ju, Xuan and Zeng, Ailing and '
'Wang, Jianan and Xu, Qiang and Zhang, Lei',
title='Human-Art: A Versatile Human-Centric Dataset '
'Bridging Natural and Artificial Scenes',
container='Proceedings of the IEEE/CVF Conference on '
'Computer Vision and Pattern Recognition',
year='2023',
homepage='https://idea-research.github.io/HumanArt/',
),
keypoint_info={
0:
dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
1:
dict(
name='left_eye',
id=1,
color=[51, 153, 255],
type='upper',
swap='right_eye'),
2:
dict(
name='right_eye',
id=2,
color=[51, 153, 255],
type='upper',
swap='left_eye'),
3:
dict(
name='left_ear',
id=3,
color=[51, 153, 255],
type='upper',
swap='right_ear'),
4:
dict(
name='right_ear',
id=4,
color=[51, 153, 255],
type='upper',
swap='left_ear'),
5:
dict(
name='left_shoulder',
id=5,
color=[0, 255, 0],
type='upper',
swap='right_shoulder'),
6:
dict(
name='right_shoulder',
id=6,
color=[255, 128, 0],
type='upper',
swap='left_shoulder'),
7:
dict(
name='left_elbow',
id=7,
color=[0, 255, 0],
type='upper',
swap='right_elbow'),
8:
dict(
name='right_elbow',
id=8,
color=[255, 128, 0],
type='upper',
swap='left_elbow'),
9:
dict(
name='left_wrist',
id=9,
color=[0, 255, 0],
type='upper',
swap='right_wrist'),
10:
dict(
name='right_wrist',
id=10,
color=[255, 128, 0],
type='upper',
swap='left_wrist'),
11:
dict(
name='left_hip',
id=11,
color=[0, 255, 0],
type='lower',
swap='right_hip'),
12:
dict(
name='right_hip',
id=12,
color=[255, 128, 0],
type='lower',
swap='left_hip'),
13:
dict(
name='left_knee',
id=13,
color=[0, 255, 0],
type='lower',
swap='right_knee'),
14:
dict(
name='right_knee',
id=14,
color=[255, 128, 0],
type='lower',
swap='left_knee'),
15:
dict(
name='left_ankle',
id=15,
color=[0, 255, 0],
type='lower',
swap='right_ankle'),
16:
dict(
name='right_ankle',
id=16,
color=[255, 128, 0],
type='lower',
swap='left_ankle')
},
skeleton_info={
0:
dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
1:
dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
2:
dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
3:
dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
4:
dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
5:
dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
6:
dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
7:
dict(
link=('left_shoulder', 'right_shoulder'),
id=7,
color=[51, 153, 255]),
8:
dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
9:
dict(
link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
10:
dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
11:
dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
12:
dict(link=('left_eye', 'right_eye'), id=12, color=[51, 153, 255]),
13:
dict(link=('nose', 'left_eye'), id=13, color=[51, 153, 255]),
14:
dict(link=('nose', 'right_eye'), id=14, color=[51, 153, 255]),
15:
dict(link=('left_eye', 'left_ear'), id=15, color=[51, 153, 255]),
16:
dict(link=('right_eye', 'right_ear'), id=16, color=[51, 153, 255]),
17:
dict(link=('left_ear', 'left_shoulder'), id=17, color=[51, 153, 255]),
18:
dict(
link=('right_ear', 'right_shoulder'), id=18, color=[51, 153, 255])
},
joint_weights=[
1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.5, 1.5, 1., 1., 1.2, 1.2, 1.5,
1.5
],
sigmas=[
0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089
])
configs/_base_/datasets/humanart21.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='Human-Art',
paper_info=dict(
author='Ju, Xuan and Zeng, Ailing and '
'Wang, Jianan and Xu, Qiang and Zhang, Lei',
title='Human-Art: A Versatile Human-Centric Dataset '
'Bridging Natural and Artificial Scenes',
container='Proceedings of the IEEE/CVF Conference on '
'Computer Vision and Pattern Recognition',
year='2023',
homepage='https://idea-research.github.io/HumanArt/',
),
keypoint_info={
0:
dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
1:
dict(
name='left_eye',
id=1,
color=[51, 153, 255],
type='upper',
swap='right_eye'),
2:
dict(
name='right_eye',
id=2,
color=[51, 153, 255],
type='upper',
swap='left_eye'),
3:
dict(
name='left_ear',
id=3,
color=[51, 153, 255],
type='upper',
swap='right_ear'),
4:
dict(
name='right_ear',
id=4,
color=[51, 153, 255],
type='upper',
swap='left_ear'),
5:
dict(
name='left_shoulder',
id=5,
color=[0, 255, 0],
type='upper',
swap='right_shoulder'),
6:
dict(
name='right_shoulder',
id=6,
color=[255, 128, 0],
type='upper',
swap='left_shoulder'),
7:
dict(
name='left_elbow',
id=7,
color=[0, 255, 0],
type='upper',
swap='right_elbow'),
8:
dict(
name='right_elbow',
id=8,
color=[255, 128, 0],
type='upper',
swap='left_elbow'),
9:
dict(
name='left_wrist',
id=9,
color=[0, 255, 0],
type='upper',
swap='right_wrist'),
10:
dict(
name='right_wrist',
id=10,
color=[255, 128, 0],
type='upper',
swap='left_wrist'),
11:
dict(
name='left_hip',
id=11,
color=[0, 255, 0],
type='lower',
swap='right_hip'),
12:
dict(
name='right_hip',
id=12,
color=[255, 128, 0],
type='lower',
swap='left_hip'),
13:
dict(
name='left_knee',
id=13,
color=[0, 255, 0],
type='lower',
swap='right_knee'),
14:
dict(
name='right_knee',
id=14,
color=[255, 128, 0],
type='lower',
swap='left_knee'),
15:
dict(
name='left_ankle',
id=15,
color=[0, 255, 0],
type='lower',
swap='right_ankle'),
16:
dict(
name='right_ankle',
id=16,
color=[255, 128, 0],
type='lower',
swap='left_ankle'),
17:
dict(
name='left_finger',
id=17,
color=[0, 255, 0],
type='lower',
swap='right_finger'),
18:
dict(
name='right_finger',
id=18,
color=[255, 128, 0],
type='lower',
swap='left_finger'),
19:
dict(
name='left_toe',
id=19,
color=[0, 255, 0],
type='lower',
swap='right_toe'),
20:
dict(
name='right_toe',
id=20,
color=[255, 128, 0],
type='lower',
swap='left_toe'),
},
skeleton_info={
0:
dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
1:
dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
2:
dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
3:
dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
4:
dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
5:
dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
6:
dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
7:
dict(
link=('left_shoulder', 'right_shoulder'),
id=7,
color=[51, 153, 255]),
8:
dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
9:
dict(
link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
10:
dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
11:
dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
12:
dict(link=('left_eye', 'right_eye'), id=12, color=[51, 153, 255]),
13:
dict(link=('nose', 'left_eye'), id=13, color=[51, 153, 255]),
14:
dict(link=('nose', 'right_eye'), id=14, color=[51, 153, 255]),
15:
dict(link=('left_eye', 'left_ear'), id=15, color=[51, 153, 255]),
16:
dict(link=('right_eye', 'right_ear'), id=16, color=[51, 153, 255]),
17:
dict(link=('left_ear', 'left_shoulder'), id=17, color=[51, 153, 255]),
18:
dict(
link=('right_ear', 'right_shoulder'), id=18, color=[51, 153, 255]),
19:
dict(link=('left_ankle', 'left_toe'), id=19, color=[0, 255, 0]),
20:
dict(link=('right_ankle', 'right_toe'), id=20, color=[255, 128, 0]),
21:
dict(link=('left_wrist', 'left_finger'), id=21, color=[0, 255, 0]),
22:
dict(link=('right_wrist', 'right_finger'), id=22, color=[255, 128, 0]),
},
joint_weights=[
1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.5, 1.5, 1., 1., 1.2, 1.2, 1.5,
1.5, 1., 1., 1., 1.
],
sigmas=[
0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089, 0.089, 0.089, 0.089,
0.089
])
configs/_base_/datasets/humanart_aic.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='humanart',
paper_info=[
dict(
author='Ju, Xuan and Zeng, Ailing and '
'Wang, Jianan and Xu, Qiang and Zhang, '
'Lei',
title='Human-Art: A Versatile Human-Centric Dataset '
'Bridging Natural and Artificial Scenes',
container='CVPR',
year='2023',
homepage='https://idea-research.github.io/HumanArt/',
),
dict(
author='Wu, Jiahong and Zheng, He and Zhao, Bo and '
'Li, Yixin and Yan, Baoming and Liang, Rui and '
'Wang, Wenjia and Zhou, Shipei and Lin, Guosen and '
'Fu, Yanwei and others',
title='Ai challenger: A large-scale dataset for going '
'deeper in image understanding',
container='arXiv',
year='2017',
homepage='https://github.com/AIChallenger/AI_Challenger_2017',
),
],
keypoint_info={
0:
dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
1:
dict(
name='left_eye',
id=1,
color=[51, 153, 255],
type='upper',
swap='right_eye'),
2:
dict(
name='right_eye',
id=2,
color=[51, 153, 255],
type='upper',
swap='left_eye'),
3:
dict(
name='left_ear',
id=3,
color=[51, 153, 255],
type='upper',
swap='right_ear'),
4:
dict(
name='right_ear',
id=4,
color=[51, 153, 255],
type='upper',
swap='left_ear'),
5:
dict(
name='left_shoulder',
id=5,
color=[0, 255, 0],
type='upper',
swap='right_shoulder'),
6:
dict(
name='right_shoulder',
id=6,
color=[255, 128, 0],
type='upper',
swap='left_shoulder'),
7:
dict(
name='left_elbow',
id=7,
color=[0, 255, 0],
type='upper',
swap='right_elbow'),
8:
dict(
name='right_elbow',
id=8,
color=[255, 128, 0],
type='upper',
swap='left_elbow'),
9:
dict(
name='left_wrist',
id=9,
color=[0, 255, 0],
type='upper',
swap='right_wrist'),
10:
dict(
name='right_wrist',
id=10,
color=[255, 128, 0],
type='upper',
swap='left_wrist'),
11:
dict(
name='left_hip',
id=11,
color=[0, 255, 0],
type='lower',
swap='right_hip'),
12:
dict(
name='right_hip',
id=12,
color=[255, 128, 0],
type='lower',
swap='left_hip'),
13:
dict(
name='left_knee',
id=13,
color=[0, 255, 0],
type='lower',
swap='right_knee'),
14:
dict(
name='right_knee',
id=14,
color=[255, 128, 0],
type='lower',
swap='left_knee'),
15:
dict(
name='left_ankle',
id=15,
color=[0, 255, 0],
type='lower',
swap='right_ankle'),
16:
dict(
name='right_ankle',
id=16,
color=[255, 128, 0],
type='lower',
swap='left_ankle'),
17:
dict(
name='head_top',
id=17,
color=[51, 153, 255],
type='upper',
swap=''),
18:
dict(name='neck', id=18, color=[51, 153, 255], type='upper', swap='')
},
skeleton_info={
0:
dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
1:
dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
2:
dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
3:
dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
4:
dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
5:
dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
6:
dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
7:
dict(
link=('left_shoulder', 'right_shoulder'),
id=7,
color=[51, 153, 255]),
8:
dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
9:
dict(
link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
10:
dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
11:
dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
12:
dict(link=('left_eye', 'right_eye'), id=12, color=[51, 153, 255]),
13:
dict(link=('nose', 'left_eye'), id=13, color=[51, 153, 255]),
14:
dict(link=('nose', 'right_eye'), id=14, color=[51, 153, 255]),
15:
dict(link=('left_eye', 'left_ear'), id=15, color=[51, 153, 255]),
16:
dict(link=('right_eye', 'right_ear'), id=16, color=[51, 153, 255]),
17:
dict(link=('left_ear', 'left_shoulder'), id=17, color=[51, 153, 255]),
18:
dict(
link=('right_ear', 'right_shoulder'), id=18, color=[51, 153, 255]),
19:
dict(link=('head_top', 'neck'), id=11, color=[51, 153, 255]),
},
joint_weights=[
1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.5, 1.5, 1., 1., 1.2, 1.2, 1.5,
1.5, 1.5
],
sigmas=[
0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089, 0.026, 0.026
])
configs/_base_/datasets/interhand2d.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='interhand2d',
paper_info=dict(
author='Moon, Gyeongsik and Yu, Shoou-I and Wen, He and '
'Shiratori, Takaaki and Lee, Kyoung Mu',
title='InterHand2.6M: A dataset and baseline for 3D '
'interacting hand pose estimation from a single RGB image',
container='arXiv',
year='2020',
homepage='https://mks0601.github.io/InterHand2.6M/',
),
keypoint_info={
0:
dict(name='thumb4', id=0, color=[255, 128, 0], type='', swap=''),
1:
dict(name='thumb3', id=1, color=[255, 128, 0], type='', swap=''),
2:
dict(name='thumb2', id=2, color=[255, 128, 0], type='', swap=''),
3:
dict(name='thumb1', id=3, color=[255, 128, 0], type='', swap=''),
4:
dict(
name='forefinger4', id=4, color=[255, 153, 255], type='', swap=''),
5:
dict(
name='forefinger3', id=5, color=[255, 153, 255], type='', swap=''),
6:
dict(
name='forefinger2', id=6, color=[255, 153, 255], type='', swap=''),
7:
dict(
name='forefinger1', id=7, color=[255, 153, 255], type='', swap=''),
8:
dict(
name='middle_finger4',
id=8,
color=[102, 178, 255],
type='',
swap=''),
9:
dict(
name='middle_finger3',
id=9,
color=[102, 178, 255],
type='',
swap=''),
10:
dict(
name='middle_finger2',
id=10,
color=[102, 178, 255],
type='',
swap=''),
11:
dict(
name='middle_finger1',
id=11,
color=[102, 178, 255],
type='',
swap=''),
12:
dict(
name='ring_finger4', id=12, color=[255, 51, 51], type='', swap=''),
13:
dict(
name='ring_finger3', id=13, color=[255, 51, 51], type='', swap=''),
14:
dict(
name='ring_finger2', id=14, color=[255, 51, 51], type='', swap=''),
15:
dict(
name='ring_finger1', id=15, color=[255, 51, 51], type='', swap=''),
16:
dict(name='pinky_finger4', id=16, color=[0, 255, 0], type='', swap=''),
17:
dict(name='pinky_finger3', id=17, color=[0, 255, 0], type='', swap=''),
18:
dict(name='pinky_finger2', id=18, color=[0, 255, 0], type='', swap=''),
19:
dict(name='pinky_finger1', id=19, color=[0, 255, 0], type='', swap=''),
20:
dict(name='wrist', id=20, color=[255, 255, 255], type='', swap='')
},
skeleton_info={
0:
dict(link=('wrist', 'thumb1'), id=0, color=[255, 128, 0]),
1:
dict(link=('thumb1', 'thumb2'), id=1, color=[255, 128, 0]),
2:
dict(link=('thumb2', 'thumb3'), id=2, color=[255, 128, 0]),
3:
dict(link=('thumb3', 'thumb4'), id=3, color=[255, 128, 0]),
4:
dict(link=('wrist', 'forefinger1'), id=4, color=[255, 153, 255]),
5:
dict(link=('forefinger1', 'forefinger2'), id=5, color=[255, 153, 255]),
6:
dict(link=('forefinger2', 'forefinger3'), id=6, color=[255, 153, 255]),
7:
dict(link=('forefinger3', 'forefinger4'), id=7, color=[255, 153, 255]),
8:
dict(link=('wrist', 'middle_finger1'), id=8, color=[102, 178, 255]),
9:
dict(
link=('middle_finger1', 'middle_finger2'),
id=9,
color=[102, 178, 255]),
10:
dict(
link=('middle_finger2', 'middle_finger3'),
id=10,
color=[102, 178, 255]),
11:
dict(
link=('middle_finger3', 'middle_finger4'),
id=11,
color=[102, 178, 255]),
12:
dict(link=('wrist', 'ring_finger1'), id=12, color=[255, 51, 51]),
13:
dict(
link=('ring_finger1', 'ring_finger2'), id=13, color=[255, 51, 51]),
14:
dict(
link=('ring_finger2', 'ring_finger3'), id=14, color=[255, 51, 51]),
15:
dict(
link=('ring_finger3', 'ring_finger4'), id=15, color=[255, 51, 51]),
16:
dict(link=('wrist', 'pinky_finger1'), id=16, color=[0, 255, 0]),
17:
dict(
link=('pinky_finger1', 'pinky_finger2'), id=17, color=[0, 255, 0]),
18:
dict(
link=('pinky_finger2', 'pinky_finger3'), id=18, color=[0, 255, 0]),
19:
dict(
link=('pinky_finger3', 'pinky_finger4'), id=19, color=[0, 255, 0])
},
joint_weights=[1.] * 21,
sigmas=[])
configs/_base_/datasets/interhand3d.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='interhand3d',
paper_info=dict(
author='Moon, Gyeongsik and Yu, Shoou-I and Wen, He and '
'Shiratori, Takaaki and Lee, Kyoung Mu',
title='InterHand2.6M: A dataset and baseline for 3D '
'interacting hand pose estimation from a single RGB image',
container='arXiv',
year='2020',
homepage='https://mks0601.github.io/InterHand2.6M/',
),
keypoint_info={
0:
dict(
name='right_thumb4',
id=0,
color=[255, 128, 0],
type='',
swap='left_thumb4'),
1:
dict(
name='right_thumb3',
id=1,
color=[255, 128, 0],
type='',
swap='left_thumb3'),
2:
dict(
name='right_thumb2',
id=2,
color=[255, 128, 0],
type='',
swap='left_thumb2'),
3:
dict(
name='right_thumb1',
id=3,
color=[255, 128, 0],
type='',
swap='left_thumb1'),
4:
dict(
name='right_forefinger4',
id=4,
color=[255, 153, 255],
type='',
swap='left_forefinger4'),
5:
dict(
name='right_forefinger3',
id=5,
color=[255, 153, 255],
type='',
swap='left_forefinger3'),
6:
dict(
name='right_forefinger2',
id=6,
color=[255, 153, 255],
type='',
swap='left_forefinger2'),
7:
dict(
name='right_forefinger1',
id=7,
color=[255, 153, 255],
type='',
swap='left_forefinger1'),
8:
dict(
name='right_middle_finger4',
id=8,
color=[102, 178, 255],
type='',
swap='left_middle_finger4'),
9:
dict(
name='right_middle_finger3',
id=9,
color=[102, 178, 255],
type='',
swap='left_middle_finger3'),
10:
dict(
name='right_middle_finger2',
id=10,
color=[102, 178, 255],
type='',
swap='left_middle_finger2'),
11:
dict(
name='right_middle_finger1',
id=11,
color=[102, 178, 255],
type='',
swap='left_middle_finger1'),
12:
dict(
name='right_ring_finger4',
id=12,
color=[255, 51, 51],
type='',
swap='left_ring_finger4'),
13:
dict(
name='right_ring_finger3',
id=13,
color=[255, 51, 51],
type='',
swap='left_ring_finger3'),
14:
dict(
name='right_ring_finger2',
id=14,
color=[255, 51, 51],
type='',
swap='left_ring_finger2'),
15:
dict(
name='right_ring_finger1',
id=15,
color=[255, 51, 51],
type='',
swap='left_ring_finger1'),
16:
dict(
name='right_pinky_finger4',
id=16,
color=[0, 255, 0],
type='',
swap='left_pinky_finger4'),
17:
dict(
name='right_pinky_finger3',
id=17,
color=[0, 255, 0],
type='',
swap='left_pinky_finger3'),
18:
dict(
name='right_pinky_finger2',
id=18,
color=[0, 255, 0],
type='',
swap='left_pinky_finger2'),
19:
dict(
name='right_pinky_finger1',
id=19,
color=[0, 255, 0],
type='',
swap='left_pinky_finger1'),
20:
dict(
name='right_wrist',
id=20,
color=[255, 255, 255],
type='',
swap='left_wrist'),
21:
dict(
name='left_thumb4',
id=21,
color=[255, 128, 0],
type='',
swap='right_thumb4'),
22:
dict(
name='left_thumb3',
id=22,
color=[255, 128, 0],
type='',
swap='right_thumb3'),
23:
dict(
name='left_thumb2',
id=23,
color=[255, 128, 0],
type='',
swap='right_thumb2'),
24:
dict(
name='left_thumb1',
id=24,
color=[255, 128, 0],
type='',
swap='right_thumb1'),
25:
dict(
name='left_forefinger4',
id=25,
color=[255, 153, 255],
type='',
swap='right_forefinger4'),
26:
dict(
name='left_forefinger3',
id=26,
color=[255, 153, 255],
type='',
swap='right_forefinger3'),
27:
dict(
name='left_forefinger2',
id=27,
color=[255, 153, 255],
type='',
swap='right_forefinger2'),
28:
dict(
name='left_forefinger1',
id=28,
color=[255, 153, 255],
type='',
swap='right_forefinger1'),
29:
dict(
name='left_middle_finger4',
id=29,
color=[102, 178, 255],
type='',
swap='right_middle_finger4'),
30:
dict(
name='left_middle_finger3',
id=30,
color=[102, 178, 255],
type='',
swap='right_middle_finger3'),
31:
dict(
name='left_middle_finger2',
id=31,
color=[102, 178, 255],
type='',
swap='right_middle_finger2'),
32:
dict(
name='left_middle_finger1',
id=32,
color=[102, 178, 255],
type='',
swap='right_middle_finger1'),
33:
dict(
name='left_ring_finger4',
id=33,
color=[255, 51, 51],
type='',
swap='right_ring_finger4'),
34:
dict(
name='left_ring_finger3',
id=34,
color=[255, 51, 51],
type='',
swap='right_ring_finger3'),
35:
dict(
name='left_ring_finger2',
id=35,
color=[255, 51, 51],
type='',
swap='right_ring_finger2'),
36:
dict(
name='left_ring_finger1',
id=36,
color=[255, 51, 51],
type='',
swap='right_ring_finger1'),
37:
dict(
name='left_pinky_finger4',
id=37,
color=[0, 255, 0],
type='',
swap='right_pinky_finger4'),
38:
dict(
name='left_pinky_finger3',
id=38,
color=[0, 255, 0],
type='',
swap='right_pinky_finger3'),
39:
dict(
name='left_pinky_finger2',
id=39,
color=[0, 255, 0],
type='',
swap='right_pinky_finger2'),
40:
dict(
name='left_pinky_finger1',
id=40,
color=[0, 255, 0],
type='',
swap='right_pinky_finger1'),
41:
dict(
name='left_wrist',
id=41,
color=[255, 255, 255],
type='',
swap='right_wrist'),
},
skeleton_info={
0:
dict(link=('right_wrist', 'right_thumb1'), id=0, color=[255, 128, 0]),
1:
dict(link=('right_thumb1', 'right_thumb2'), id=1, color=[255, 128, 0]),
2:
dict(link=('right_thumb2', 'right_thumb3'), id=2, color=[255, 128, 0]),
3:
dict(link=('right_thumb3', 'right_thumb4'), id=3, color=[255, 128, 0]),
4:
dict(
link=('right_wrist', 'right_forefinger1'),
id=4,
color=[255, 153, 255]),
5:
dict(
link=('right_forefinger1', 'right_forefinger2'),
id=5,
color=[255, 153, 255]),
6:
dict(
link=('right_forefinger2', 'right_forefinger3'),
id=6,
color=[255, 153, 255]),
7:
dict(
link=('right_forefinger3', 'right_forefinger4'),
id=7,
color=[255, 153, 255]),
8:
dict(
link=('right_wrist', 'right_middle_finger1'),
id=8,
color=[102, 178, 255]),
9:
dict(
link=('right_middle_finger1', 'right_middle_finger2'),
id=9,
color=[102, 178, 255]),
10:
dict(
link=('right_middle_finger2', 'right_middle_finger3'),
id=10,
color=[102, 178, 255]),
11:
dict(
link=('right_middle_finger3', 'right_middle_finger4'),
id=11,
color=[102, 178, 255]),
12:
dict(
link=('right_wrist', 'right_ring_finger1'),
id=12,
color=[255, 51, 51]),
13:
dict(
link=('right_ring_finger1', 'right_ring_finger2'),
id=13,
color=[255, 51, 51]),
14:
dict(
link=('right_ring_finger2', 'right_ring_finger3'),
id=14,
color=[255, 51, 51]),
15:
dict(
link=('right_ring_finger3', 'right_ring_finger4'),
id=15,
color=[255, 51, 51]),
16:
dict(
link=('right_wrist', 'right_pinky_finger1'),
id=16,
color=[0, 255, 0]),
17:
dict(
link=('right_pinky_finger1', 'right_pinky_finger2'),
id=17,
color=[0, 255, 0]),
18:
dict(
link=('right_pinky_finger2', 'right_pinky_finger3'),
id=18,
color=[0, 255, 0]),
19:
dict(
link=('right_pinky_finger3', 'right_pinky_finger4'),
id=19,
color=[0, 255, 0]),
20:
dict(link=('left_wrist', 'left_thumb1'), id=20, color=[255, 128, 0]),
21:
dict(link=('left_thumb1', 'left_thumb2'), id=21, color=[255, 128, 0]),
22:
dict(link=('left_thumb2', 'left_thumb3'), id=22, color=[255, 128, 0]),
23:
dict(link=('left_thumb3', 'left_thumb4'), id=23, color=[255, 128, 0]),
24:
dict(
link=('left_wrist', 'left_forefinger1'),
id=24,
color=[255, 153, 255]),
25:
dict(
link=('left_forefinger1', 'left_forefinger2'),
id=25,
color=[255, 153, 255]),
26:
dict(
link=('left_forefinger2', 'left_forefinger3'),
id=26,
color=[255, 153, 255]),
27:
dict(
link=('left_forefinger3', 'left_forefinger4'),
id=27,
color=[255, 153, 255]),
28:
dict(
link=('left_wrist', 'left_middle_finger1'),
id=28,
color=[102, 178, 255]),
29:
dict(
link=('left_middle_finger1', 'left_middle_finger2'),
id=29,
color=[102, 178, 255]),
30:
dict(
link=('left_middle_finger2', 'left_middle_finger3'),
id=30,
color=[102, 178, 255]),
31:
dict(
link=('left_middle_finger3', 'left_middle_finger4'),
id=31,
color=[102, 178, 255]),
32:
dict(
link=('left_wrist', 'left_ring_finger1'),
id=32,
color=[255, 51, 51]),
33:
dict(
link=('left_ring_finger1', 'left_ring_finger2'),
id=33,
color=[255, 51, 51]),
34:
dict(
link=('left_ring_finger2', 'left_ring_finger3'),
id=34,
color=[255, 51, 51]),
35:
dict(
link=('left_ring_finger3', 'left_ring_finger4'),
id=35,
color=[255, 51, 51]),
36:
dict(
link=('left_wrist', 'left_pinky_finger1'),
id=36,
color=[0, 255, 0]),
37:
dict(
link=('left_pinky_finger1', 'left_pinky_finger2'),
id=37,
color=[0, 255, 0]),
38:
dict(
link=('left_pinky_finger2', 'left_pinky_finger3'),
id=38,
color=[0, 255, 0]),
39:
dict(
link=('left_pinky_finger3', 'left_pinky_finger4'),
id=39,
color=[0, 255, 0]),
},
joint_weights=[1.] * 42,
sigmas=[])
configs/_base_/datasets/jhmdb.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='jhmdb',
paper_info=dict(
author='H. Jhuang and J. Gall and S. Zuffi and '
'C. Schmid and M. J. Black',
title='Towards understanding action recognition',
container='International Conf. on Computer Vision (ICCV)',
year='2013',
homepage='http://jhmdb.is.tue.mpg.de/dataset',
),
keypoint_info={
0:
dict(name='neck', id=0, color=[255, 128, 0], type='upper', swap=''),
1:
dict(name='belly', id=1, color=[255, 128, 0], type='upper', swap=''),
2:
dict(name='head', id=2, color=[255, 128, 0], type='upper', swap=''),
3:
dict(
name='right_shoulder',
id=3,
color=[0, 255, 0],
type='upper',
swap='left_shoulder'),
4:
dict(
name='left_shoulder',
id=4,
color=[0, 255, 0],
type='upper',
swap='right_shoulder'),
5:
dict(
name='right_hip',
id=5,
color=[0, 255, 0],
type='lower',
swap='left_hip'),
6:
dict(
name='left_hip',
id=6,
color=[51, 153, 255],
type='lower',
swap='right_hip'),
7:
dict(
name='right_elbow',
id=7,
color=[51, 153, 255],
type='upper',
swap='left_elbow'),
8:
dict(
name='left_elbow',
id=8,
color=[51, 153, 255],
type='upper',
swap='right_elbow'),
9:
dict(
name='right_knee',
id=9,
color=[51, 153, 255],
type='lower',
swap='left_knee'),
10:
dict(
name='left_knee',
id=10,
color=[255, 128, 0],
type='lower',
swap='right_knee'),
11:
dict(
name='right_wrist',
id=11,
color=[255, 128, 0],
type='upper',
swap='left_wrist'),
12:
dict(
name='left_wrist',
id=12,
color=[255, 128, 0],
type='upper',
swap='right_wrist'),
13:
dict(
name='right_ankle',
id=13,
color=[0, 255, 0],
type='lower',
swap='left_ankle'),
14:
dict(
name='left_ankle',
id=14,
color=[0, 255, 0],
type='lower',
swap='right_ankle')
},
skeleton_info={
0: dict(link=('right_ankle', 'right_knee'), id=0, color=[255, 128, 0]),
1: dict(link=('right_knee', 'right_hip'), id=1, color=[255, 128, 0]),
2: dict(link=('right_hip', 'belly'), id=2, color=[255, 128, 0]),
3: dict(link=('belly', 'left_hip'), id=3, color=[0, 255, 0]),
4: dict(link=('left_hip', 'left_knee'), id=4, color=[0, 255, 0]),
5: dict(link=('left_knee', 'left_ankle'), id=5, color=[0, 255, 0]),
6: dict(link=('belly', 'neck'), id=6, color=[51, 153, 255]),
7: dict(link=('neck', 'head'), id=7, color=[51, 153, 255]),
8: dict(link=('neck', 'right_shoulder'), id=8, color=[255, 128, 0]),
9: dict(
link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
10:
dict(link=('right_elbow', 'right_wrist'), id=10, color=[255, 128, 0]),
11: dict(link=('neck', 'left_shoulder'), id=11, color=[0, 255, 0]),
12:
dict(link=('left_shoulder', 'left_elbow'), id=12, color=[0, 255, 0]),
13: dict(link=('left_elbow', 'left_wrist'), id=13, color=[0, 255, 0])
},
joint_weights=[
1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.2, 1.2, 1.5, 1.5, 1.5, 1.5
],
# Adapted from COCO dataset.
sigmas=[
0.025, 0.107, 0.025, 0.079, 0.079, 0.107, 0.107, 0.072, 0.072, 0.087,
0.087, 0.062, 0.062, 0.089, 0.089
])
configs/_base_/datasets/lapa.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='lapa',
paper_info=dict(
author='Liu, Yinglu and Shi, Hailin and Shen, Hao and Si, '
'Yue and Wang, Xiaobo and Mei, Tao',
title='A New Dataset and Boundary-Attention Semantic '
'Segmentation for Face Parsing.',
container='Proceedings of the AAAI Conference on '
'Artificial Intelligence 2020',
year='2020',
homepage='https://github.com/JDAI-CV/lapa-dataset',
),
keypoint_info={
0:
dict(name='kpt-0', id=0, color=[255, 0, 0], type='', swap='kpt-32'),
1:
dict(name='kpt-1', id=1, color=[255, 0, 0], type='', swap='kpt-31'),
2:
dict(name='kpt-2', id=2, color=[255, 0, 0], type='', swap='kpt-30'),
3:
dict(name='kpt-3', id=3, color=[255, 0, 0], type='', swap='kpt-29'),
4:
dict(name='kpt-4', id=4, color=[255, 0, 0], type='', swap='kpt-28'),
5:
dict(name='kpt-5', id=5, color=[255, 0, 0], type='', swap='kpt-27'),
6:
dict(name='kpt-6', id=6, color=[255, 0, 0], type='', swap='kpt-26'),
7:
dict(name='kpt-7', id=7, color=[255, 0, 0], type='', swap='kpt-25'),
8:
dict(name='kpt-8', id=8, color=[255, 0, 0], type='', swap='kpt-24'),
9:
dict(name='kpt-9', id=9, color=[255, 0, 0], type='', swap='kpt-23'),
10:
dict(name='kpt-10', id=10, color=[255, 0, 0], type='', swap='kpt-22'),
11:
dict(name='kpt-11', id=11, color=[255, 0, 0], type='', swap='kpt-21'),
12:
dict(name='kpt-12', id=12, color=[255, 0, 0], type='', swap='kpt-20'),
13:
dict(name='kpt-13', id=13, color=[255, 0, 0], type='', swap='kpt-19'),
14:
dict(name='kpt-14', id=14, color=[255, 0, 0], type='', swap='kpt-18'),
15:
dict(name='kpt-15', id=15, color=[255, 0, 0], type='', swap='kpt-17'),
16:
dict(name='kpt-16', id=16, color=[255, 0, 0], type='', swap=''),
17:
dict(name='kpt-17', id=17, color=[255, 0, 0], type='', swap='kpt-15'),
18:
dict(name='kpt-18', id=18, color=[255, 0, 0], type='', swap='kpt-14'),
19:
dict(name='kpt-19', id=19, color=[255, 0, 0], type='', swap='kpt-13'),
20:
dict(name='kpt-20', id=20, color=[255, 0, 0], type='', swap='kpt-12'),
21:
dict(name='kpt-21', id=21, color=[255, 0, 0], type='', swap='kpt-11'),
22:
dict(name='kpt-22', id=22, color=[255, 0, 0], type='', swap='kpt-10'),
23:
dict(name='kpt-23', id=23, color=[255, 0, 0], type='', swap='kpt-9'),
24:
dict(name='kpt-24', id=24, color=[255, 0, 0], type='', swap='kpt-8'),
25:
dict(name='kpt-25', id=25, color=[255, 0, 0], type='', swap='kpt-7'),
26:
dict(name='kpt-26', id=26, color=[255, 0, 0], type='', swap='kpt-6'),
27:
dict(name='kpt-27', id=27, color=[255, 0, 0], type='', swap='kpt-5'),
28:
dict(name='kpt-28', id=28, color=[255, 0, 0], type='', swap='kpt-4'),
29:
dict(name='kpt-29', id=29, color=[255, 0, 0], type='', swap='kpt-3'),
30:
dict(name='kpt-30', id=30, color=[255, 0, 0], type='', swap='kpt-2'),
31:
dict(name='kpt-31', id=31, color=[255, 0, 0], type='', swap='kpt-1'),
32:
dict(name='kpt-32', id=32, color=[255, 0, 0], type='', swap='kpt-0'),
33:
dict(name='kpt-33', id=33, color=[255, 0, 0], type='', swap='kpt-46'),
34:
dict(name='kpt-34', id=34, color=[255, 0, 0], type='', swap='kpt-45'),
35:
dict(name='kpt-35', id=35, color=[255, 0, 0], type='', swap='kpt-44'),
36:
dict(name='kpt-36', id=36, color=[255, 0, 0], type='', swap='kpt-43'),
37:
dict(name='kpt-37', id=37, color=[255, 0, 0], type='', swap='kpt-42'),
38:
dict(name='kpt-38', id=38, color=[255, 0, 0], type='', swap='kpt-50'),
39:
dict(name='kpt-39', id=39, color=[255, 0, 0], type='', swap='kpt-49'),
40:
dict(name='kpt-40', id=40, color=[255, 0, 0], type='', swap='kpt-48'),
41:
dict(name='kpt-41', id=41, color=[255, 0, 0], type='', swap='kpt-47'),
42:
dict(name='kpt-42', id=42, color=[255, 0, 0], type='', swap='kpt-37'),
43:
dict(name='kpt-43', id=43, color=[255, 0, 0], type='', swap='kpt-36'),
44:
dict(name='kpt-44', id=44, color=[255, 0, 0], type='', swap='kpt-35'),
45:
dict(name='kpt-45', id=45, color=[255, 0, 0], type='', swap='kpt-34'),
46:
dict(name='kpt-46', id=46, color=[255, 0, 0], type='', swap='kpt-33'),
47:
dict(name='kpt-47', id=47, color=[255, 0, 0], type='', swap='kpt-41'),
48:
dict(name='kpt-48', id=48, color=[255, 0, 0], type='', swap='kpt-40'),
49:
dict(name='kpt-49', id=49, color=[255, 0, 0], type='', swap='kpt-39'),
50:
dict(name='kpt-50', id=50, color=[255, 0, 0], type='', swap='kpt-38'),
51:
dict(name='kpt-51', id=51, color=[255, 0, 0], type='', swap=''),
52:
dict(name='kpt-52', id=52, color=[255, 0, 0], type='', swap=''),
53:
dict(name='kpt-53', id=53, color=[255, 0, 0], type='', swap=''),
54:
dict(name='kpt-54', id=54, color=[255, 0, 0], type='', swap=''),
55:
dict(name='kpt-55', id=55, color=[255, 0, 0], type='', swap='kpt-65'),
56:
dict(name='kpt-56', id=56, color=[255, 0, 0], type='', swap='kpt-64'),
57:
dict(name='kpt-57', id=57, color=[255, 0, 0], type='', swap='kpt-63'),
58:
dict(name='kpt-58', id=58, color=[255, 0, 0], type='', swap='kpt-62'),
59:
dict(name='kpt-59', id=59, color=[255, 0, 0], type='', swap='kpt-61'),
60:
dict(name='kpt-60', id=60, color=[255, 0, 0], type='', swap=''),
61:
dict(name='kpt-61', id=61, color=[255, 0, 0], type='', swap='kpt-59'),
62:
dict(name='kpt-62', id=62, color=[255, 0, 0], type='', swap='kpt-58'),
63:
dict(name='kpt-63', id=63, color=[255, 0, 0], type='', swap='kpt-57'),
64:
dict(name='kpt-64', id=64, color=[255, 0, 0], type='', swap='kpt-56'),
65:
dict(name='kpt-65', id=65, color=[255, 0, 0], type='', swap='kpt-55'),
66:
dict(name='kpt-66', id=66, color=[255, 0, 0], type='', swap='kpt-79'),
67:
dict(name='kpt-67', id=67, color=[255, 0, 0], type='', swap='kpt-78'),
68:
dict(name='kpt-68', id=68, color=[255, 0, 0], type='', swap='kpt-77'),
69:
dict(name='kpt-69', id=69, color=[255, 0, 0], type='', swap='kpt-76'),
70:
dict(name='kpt-70', id=70, color=[255, 0, 0], type='', swap='kpt-75'),
71:
dict(name='kpt-71', id=71, color=[255, 0, 0], type='', swap='kpt-82'),
72:
dict(name='kpt-72', id=72, color=[255, 0, 0], type='', swap='kpt-81'),
73:
dict(name='kpt-73', id=73, color=[255, 0, 0], type='', swap='kpt-80'),
74:
dict(name='kpt-74', id=74, color=[255, 0, 0], type='', swap='kpt-83'),
75:
dict(name='kpt-75', id=75, color=[255, 0, 0], type='', swap='kpt-70'),
76:
dict(name='kpt-76', id=76, color=[255, 0, 0], type='', swap='kpt-69'),
77:
dict(name='kpt-77', id=77, color=[255, 0, 0], type='', swap='kpt-68'),
78:
dict(name='kpt-78', id=78, color=[255, 0, 0], type='', swap='kpt-67'),
79:
dict(name='kpt-79', id=79, color=[255, 0, 0], type='', swap='kpt-66'),
80:
dict(name='kpt-80', id=80, color=[255, 0, 0], type='', swap='kpt-73'),
81:
dict(name='kpt-81', id=81, color=[255, 0, 0], type='', swap='kpt-72'),
82:
dict(name='kpt-82', id=82, color=[255, 0, 0], type='', swap='kpt-71'),
83:
dict(name='kpt-83', id=83, color=[255, 0, 0], type='', swap='kpt-74'),
84:
dict(name='kpt-84', id=84, color=[255, 0, 0], type='', swap='kpt-90'),
85:
dict(name='kpt-85', id=85, color=[255, 0, 0], type='', swap='kpt-89'),
86:
dict(name='kpt-86', id=86, color=[255, 0, 0], type='', swap='kpt-88'),
87:
dict(name='kpt-87', id=87, color=[255, 0, 0], type='', swap=''),
88:
dict(name='kpt-88', id=88, color=[255, 0, 0], type='', swap='kpt-86'),
89:
dict(name='kpt-89', id=89, color=[255, 0, 0], type='', swap='kpt-85'),
90:
dict(name='kpt-90', id=90, color=[255, 0, 0], type='', swap='kpt-84'),
91:
dict(name='kpt-91', id=91, color=[255, 0, 0], type='', swap='kpt-95'),
92:
dict(name='kpt-92', id=92, color=[255, 0, 0], type='', swap='kpt-94'),
93:
dict(name='kpt-93', id=93, color=[255, 0, 0], type='', swap=''),
94:
dict(name='kpt-94', id=94, color=[255, 0, 0], type='', swap='kpt-92'),
95:
dict(name='kpt-95', id=95, color=[255, 0, 0], type='', swap='kpt-91'),
96:
dict(name='kpt-96', id=96, color=[255, 0, 0], type='', swap='kpt-100'),
97:
dict(name='kpt-97', id=97, color=[255, 0, 0], type='', swap='kpt-99'),
98:
dict(name='kpt-98', id=98, color=[255, 0, 0], type='', swap=''),
99:
dict(name='kpt-99', id=99, color=[255, 0, 0], type='', swap='kpt-97'),
100:
dict(
name='kpt-100', id=100, color=[255, 0, 0], type='', swap='kpt-96'),
101:
dict(
name='kpt-101', id=101, color=[255, 0, 0], type='',
swap='kpt-103'),
102:
dict(name='kpt-102', id=102, color=[255, 0, 0], type='', swap=''),
103:
dict(
name='kpt-103', id=103, color=[255, 0, 0], type='',
swap='kpt-101'),
104:
dict(
name='kpt-104', id=104, color=[255, 0, 0], type='',
swap='kpt-105'),
105:
dict(
name='kpt-105', id=105, color=[255, 0, 0], type='', swap='kpt-104')
},
skeleton_info={},
joint_weights=[
0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8,
0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8,
0.8, 0.8, 0.8, 0.8, 0.8, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 2.0, 2.0, 2.0, 2.0,
2.0, 2.0, 2.0, 2.0, 1.0, 2.0, 2.0, 2.0, 2.0, 2.0, 2.0, 2.0, 2.0, 1.0,
1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5,
1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.0, 1.0
],
sigmas=[])
configs/_base_/datasets/locust.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='locust',
paper_info=dict(
author='Graving, Jacob M and Chae, Daniel and Naik, Hemal and '
'Li, Liang and Koger, Benjamin and Costelloe, Blair R and '
'Couzin, Iain D',
title='DeepPoseKit, a software toolkit for fast and robust '
'animal pose estimation using deep learning',
container='Elife',
year='2019',
homepage='https://github.com/jgraving/DeepPoseKit-Data',
),
keypoint_info={
0:
dict(name='head', id=0, color=[255, 255, 255], type='', swap=''),
1:
dict(name='neck', id=1, color=[255, 255, 255], type='', swap=''),
2:
dict(name='thorax', id=2, color=[255, 255, 255], type='', swap=''),
3:
dict(name='abdomen1', id=3, color=[255, 255, 255], type='', swap=''),
4:
dict(name='abdomen2', id=4, color=[255, 255, 255], type='', swap=''),
5:
dict(
name='anttipL',
id=5,
color=[255, 255, 255],
type='',
swap='anttipR'),
6:
dict(
name='antbaseL',
id=6,
color=[255, 255, 255],
type='',
swap='antbaseR'),
7:
dict(name='eyeL', id=7, color=[255, 255, 255], type='', swap='eyeR'),
8:
dict(
name='forelegL1',
id=8,
color=[255, 255, 255],
type='',
swap='forelegR1'),
9:
dict(
name='forelegL2',
id=9,
color=[255, 255, 255],
type='',
swap='forelegR2'),
10:
dict(
name='forelegL3',
id=10,
color=[255, 255, 255],
type='',
swap='forelegR3'),
11:
dict(
name='forelegL4',
id=11,
color=[255, 255, 255],
type='',
swap='forelegR4'),
12:
dict(
name='midlegL1',
id=12,
color=[255, 255, 255],
type='',
swap='midlegR1'),
13:
dict(
name='midlegL2',
id=13,
color=[255, 255, 255],
type='',
swap='midlegR2'),
14:
dict(
name='midlegL3',
id=14,
color=[255, 255, 255],
type='',
swap='midlegR3'),
15:
dict(
name='midlegL4',
id=15,
color=[255, 255, 255],
type='',
swap='midlegR4'),
16:
dict(
name='hindlegL1',
id=16,
color=[255, 255, 255],
type='',
swap='hindlegR1'),
17:
dict(
name='hindlegL2',
id=17,
color=[255, 255, 255],
type='',
swap='hindlegR2'),
18:
dict(
name='hindlegL3',
id=18,
color=[255, 255, 255],
type='',
swap='hindlegR3'),
19:
dict(
name='hindlegL4',
id=19,
color=[255, 255, 255],
type='',
swap='hindlegR4'),
20:
dict(
name='anttipR',
id=20,
color=[255, 255, 255],
type='',
swap='anttipL'),
21:
dict(
name='antbaseR',
id=21,
color=[255, 255, 255],
type='',
swap='antbaseL'),
22:
dict(name='eyeR', id=22, color=[255, 255, 255], type='', swap='eyeL'),
23:
dict(
name='forelegR1',
id=23,
color=[255, 255, 255],
type='',
swap='forelegL1'),
24:
dict(
name='forelegR2',
id=24,
color=[255, 255, 255],
type='',
swap='forelegL2'),
25:
dict(
name='forelegR3',
id=25,
color=[255, 255, 255],
type='',
swap='forelegL3'),
26:
dict(
name='forelegR4',
id=26,
color=[255, 255, 255],
type='',
swap='forelegL4'),
27:
dict(
name='midlegR1',
id=27,
color=[255, 255, 255],
type='',
swap='midlegL1'),
28:
dict(
name='midlegR2',
id=28,
color=[255, 255, 255],
type='',
swap='midlegL2'),
29:
dict(
name='midlegR3',
id=29,
color=[255, 255, 255],
type='',
swap='midlegL3'),
30:
dict(
name='midlegR4',
id=30,
color=[255, 255, 255],
type='',
swap='midlegL4'),
31:
dict(
name='hindlegR1',
id=31,
color=[255, 255, 255],
type='',
swap='hindlegL1'),
32:
dict(
name='hindlegR2',
id=32,
color=[255, 255, 255],
type='',
swap='hindlegL2'),
33:
dict(
name='hindlegR3',
id=33,
color=[255, 255, 255],
type='',
swap='hindlegL3'),
34:
dict(
name='hindlegR4',
id=34,
color=[255, 255, 255],
type='',
swap='hindlegL4')
},
skeleton_info={
0: dict(link=('neck', 'head'), id=0, color=[255, 255, 255]),
1: dict(link=('thorax', 'neck'), id=1, color=[255, 255, 255]),
2: dict(link=('abdomen1', 'thorax'), id=2, color=[255, 255, 255]),
3: dict(link=('abdomen2', 'abdomen1'), id=3, color=[255, 255, 255]),
4: dict(link=('antbaseL', 'anttipL'), id=4, color=[255, 255, 255]),
5: dict(link=('eyeL', 'antbaseL'), id=5, color=[255, 255, 255]),
6: dict(link=('forelegL2', 'forelegL1'), id=6, color=[255, 255, 255]),
7: dict(link=('forelegL3', 'forelegL2'), id=7, color=[255, 255, 255]),
8: dict(link=('forelegL4', 'forelegL3'), id=8, color=[255, 255, 255]),
9: dict(link=('midlegL2', 'midlegL1'), id=9, color=[255, 255, 255]),
10: dict(link=('midlegL3', 'midlegL2'), id=10, color=[255, 255, 255]),
11: dict(link=('midlegL4', 'midlegL3'), id=11, color=[255, 255, 255]),
12:
dict(link=('hindlegL2', 'hindlegL1'), id=12, color=[255, 255, 255]),
13:
dict(link=('hindlegL3', 'hindlegL2'), id=13, color=[255, 255, 255]),
14:
dict(link=('hindlegL4', 'hindlegL3'), id=14, color=[255, 255, 255]),
15: dict(link=('antbaseR', 'anttipR'), id=15, color=[255, 255, 255]),
16: dict(link=('eyeR', 'antbaseR'), id=16, color=[255, 255, 255]),
17:
dict(link=('forelegR2', 'forelegR1'), id=17, color=[255, 255, 255]),
18:
dict(link=('forelegR3', 'forelegR2'), id=18, color=[255, 255, 255]),
19:
dict(link=('forelegR4', 'forelegR3'), id=19, color=[255, 255, 255]),
20: dict(link=('midlegR2', 'midlegR1'), id=20, color=[255, 255, 255]),
21: dict(link=('midlegR3', 'midlegR2'), id=21, color=[255, 255, 255]),
22: dict(link=('midlegR4', 'midlegR3'), id=22, color=[255, 255, 255]),
23:
dict(link=('hindlegR2', 'hindlegR1'), id=23, color=[255, 255, 255]),
24:
dict(link=('hindlegR3', 'hindlegR2'), id=24, color=[255, 255, 255]),
25:
dict(link=('hindlegR4', 'hindlegR3'), id=25, color=[255, 255, 255])
},
joint_weights=[1.] * 35,
sigmas=[])
configs/_base_/datasets/macaque.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='macaque',
paper_info=dict(
author='Labuguen, Rollyn and Matsumoto, Jumpei and '
'Negrete, Salvador and Nishimaru, Hiroshi and '
'Nishijo, Hisao and Takada, Masahiko and '
'Go, Yasuhiro and Inoue, Ken-ichi and Shibata, Tomohiro',
title='MacaquePose: A novel "in the wild" macaque monkey pose dataset '
'for markerless motion capture',
container='bioRxiv',
year='2020',
homepage='http://www.pri.kyoto-u.ac.jp/datasets/'
'macaquepose/index.html',
),
keypoint_info={
0:
dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
1:
dict(
name='left_eye',
id=1,
color=[51, 153, 255],
type='upper',
swap='right_eye'),
2:
dict(
name='right_eye',
id=2,
color=[51, 153, 255],
type='upper',
swap='left_eye'),
3:
dict(
name='left_ear',
id=3,
color=[51, 153, 255],
type='upper',
swap='right_ear'),
4:
dict(
name='right_ear',
id=4,
color=[51, 153, 255],
type='upper',
swap='left_ear'),
5:
dict(
name='left_shoulder',
id=5,
color=[0, 255, 0],
type='upper',
swap='right_shoulder'),
6:
dict(
name='right_shoulder',
id=6,
color=[255, 128, 0],
type='upper',
swap='left_shoulder'),
7:
dict(
name='left_elbow',
id=7,
color=[0, 255, 0],
type='upper',
swap='right_elbow'),
8:
dict(
name='right_elbow',
id=8,
color=[255, 128, 0],
type='upper',
swap='left_elbow'),
9:
dict(
name='left_wrist',
id=9,
color=[0, 255, 0],
type='upper',
swap='right_wrist'),
10:
dict(
name='right_wrist',
id=10,
color=[255, 128, 0],
type='upper',
swap='left_wrist'),
11:
dict(
name='left_hip',
id=11,
color=[0, 255, 0],
type='lower',
swap='right_hip'),
12:
dict(
name='right_hip',
id=12,
color=[255, 128, 0],
type='lower',
swap='left_hip'),
13:
dict(
name='left_knee',
id=13,
color=[0, 255, 0],
type='lower',
swap='right_knee'),
14:
dict(
name='right_knee',
id=14,
color=[255, 128, 0],
type='lower',
swap='left_knee'),
15:
dict(
name='left_ankle',
id=15,
color=[0, 255, 0],
type='lower',
swap='right_ankle'),
16:
dict(
name='right_ankle',
id=16,
color=[255, 128, 0],
type='lower',
swap='left_ankle')
},
skeleton_info={
0:
dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
1:
dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
2:
dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
3:
dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
4:
dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
5:
dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
6:
dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
7:
dict(
link=('left_shoulder', 'right_shoulder'),
id=7,
color=[51, 153, 255]),
8:
dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
9:
dict(
link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
10:
dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
11:
dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
12:
dict(link=('left_eye', 'right_eye'), id=12, color=[51, 153, 255]),
13:
dict(link=('nose', 'left_eye'), id=13, color=[51, 153, 255]),
14:
dict(link=('nose', 'right_eye'), id=14, color=[51, 153, 255]),
15:
dict(link=('left_eye', 'left_ear'), id=15, color=[51, 153, 255]),
16:
dict(link=('right_eye', 'right_ear'), id=16, color=[51, 153, 255]),
17:
dict(link=('left_ear', 'left_shoulder'), id=17, color=[51, 153, 255]),
18:
dict(
link=('right_ear', 'right_shoulder'), id=18, color=[51, 153, 255])
},
joint_weights=[
1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.5, 1.5, 1., 1., 1.2, 1.2, 1.5,
1.5
],
sigmas=[
0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089
])
configs/_base_/datasets/mhp.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='mhp',
paper_info=dict(
author='Zhao, Jian and Li, Jianshu and Cheng, Yu and '
'Sim, Terence and Yan, Shuicheng and Feng, Jiashi',
title='Understanding humans in crowded scenes: '
'Deep nested adversarial learning and a '
'new benchmark for multi-human parsing',
container='Proceedings of the 26th ACM '
'international conference on Multimedia',
year='2018',
homepage='https://lv-mhp.github.io/dataset',
),
keypoint_info={
0:
dict(
name='right_ankle',
id=0,
color=[255, 128, 0],
type='lower',
swap='left_ankle'),
1:
dict(
name='right_knee',
id=1,
color=[255, 128, 0],
type='lower',
swap='left_knee'),
2:
dict(
name='right_hip',
id=2,
color=[255, 128, 0],
type='lower',
swap='left_hip'),
3:
dict(
name='left_hip',
id=3,
color=[0, 255, 0],
type='lower',
swap='right_hip'),
4:
dict(
name='left_knee',
id=4,
color=[0, 255, 0],
type='lower',
swap='right_knee'),
5:
dict(
name='left_ankle',
id=5,
color=[0, 255, 0],
type='lower',
swap='right_ankle'),
6:
dict(name='pelvis', id=6, color=[51, 153, 255], type='lower', swap=''),
7:
dict(name='thorax', id=7, color=[51, 153, 255], type='upper', swap=''),
8:
dict(
name='upper_neck',
id=8,
color=[51, 153, 255],
type='upper',
swap=''),
9:
dict(
name='head_top', id=9, color=[51, 153, 255], type='upper',
swap=''),
10:
dict(
name='right_wrist',
id=10,
color=[255, 128, 0],
type='upper',
swap='left_wrist'),
11:
dict(
name='right_elbow',
id=11,
color=[255, 128, 0],
type='upper',
swap='left_elbow'),
12:
dict(
name='right_shoulder',
id=12,
color=[255, 128, 0],
type='upper',
swap='left_shoulder'),
13:
dict(
name='left_shoulder',
id=13,
color=[0, 255, 0],
type='upper',
swap='right_shoulder'),
14:
dict(
name='left_elbow',
id=14,
color=[0, 255, 0],
type='upper',
swap='right_elbow'),
15:
dict(
name='left_wrist',
id=15,
color=[0, 255, 0],
type='upper',
swap='right_wrist')
},
skeleton_info={
0:
dict(link=('right_ankle', 'right_knee'), id=0, color=[255, 128, 0]),
1:
dict(link=('right_knee', 'right_hip'), id=1, color=[255, 128, 0]),
2:
dict(link=('right_hip', 'pelvis'), id=2, color=[255, 128, 0]),
3:
dict(link=('pelvis', 'left_hip'), id=3, color=[0, 255, 0]),
4:
dict(link=('left_hip', 'left_knee'), id=4, color=[0, 255, 0]),
5:
dict(link=('left_knee', 'left_ankle'), id=5, color=[0, 255, 0]),
6:
dict(link=('pelvis', 'thorax'), id=6, color=[51, 153, 255]),
7:
dict(link=('thorax', 'upper_neck'), id=7, color=[51, 153, 255]),
8:
dict(link=('upper_neck', 'head_top'), id=8, color=[51, 153, 255]),
9:
dict(link=('upper_neck', 'right_shoulder'), id=9, color=[255, 128, 0]),
10:
dict(
link=('right_shoulder', 'right_elbow'), id=10, color=[255, 128,
0]),
11:
dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
12:
dict(link=('upper_neck', 'left_shoulder'), id=12, color=[0, 255, 0]),
13:
dict(link=('left_shoulder', 'left_elbow'), id=13, color=[0, 255, 0]),
14:
dict(link=('left_elbow', 'left_wrist'), id=14, color=[0, 255, 0])
},
joint_weights=[
1.5, 1.2, 1., 1., 1.2, 1.5, 1., 1., 1., 1., 1.5, 1.2, 1., 1., 1.2, 1.5
],
# Adapted from COCO dataset.
sigmas=[
0.089, 0.083, 0.107, 0.107, 0.083, 0.089, 0.026, 0.026, 0.026, 0.026,
0.062, 0.072, 0.179, 0.179, 0.072, 0.062
])
configs/_base_/datasets/mpi_inf_3dhp.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='mpi_inf_3dhp',
paper_info=dict(
author='ehta, Dushyant and Rhodin, Helge and Casas, Dan and '
'Fua, Pascal and Sotnychenko, Oleksandr and Xu, Weipeng and '
'Theobalt, Christian',
title='Monocular 3D Human Pose Estimation In The Wild Using Improved '
'CNN Supervision',
container='2017 international conference on 3D vision (3DV)',
year='2017',
homepage='http://gvv.mpi-inf.mpg.de/3dhp-dataset',
),
keypoint_info={
0:
dict(
name='head_top', id=0, color=[51, 153, 255], type='upper',
swap=''),
1:
dict(name='neck', id=1, color=[51, 153, 255], type='upper', swap=''),
2:
dict(
name='right_shoulder',
id=2,
color=[255, 128, 0],
type='upper',
swap='left_shoulder'),
3:
dict(
name='right_elbow',
id=3,
color=[255, 128, 0],
type='upper',
swap='left_elbow'),
4:
dict(
name='right_wrist',
id=4,
color=[255, 128, 0],
type='upper',
swap='left_wrist'),
5:
dict(
name='left_shoulder',
id=5,
color=[0, 255, 0],
type='upper',
swap='right_shoulder'),
6:
dict(
name='left_elbow',
id=6,
color=[0, 255, 0],
type='upper',
swap='right_elbow'),
7:
dict(
name='left_wrist',
id=7,
color=[0, 255, 0],
type='upper',
swap='right_wrist'),
8:
dict(
name='right_hip',
id=8,
color=[255, 128, 0],
type='lower',
swap='left_hip'),
9:
dict(
name='right_knee',
id=9,
color=[255, 128, 0],
type='lower',
swap='left_knee'),
10:
dict(
name='right_ankle',
id=10,
color=[255, 128, 0],
type='lower',
swap='left_ankle'),
11:
dict(
name='left_hip',
id=11,
color=[0, 255, 0],
type='lower',
swap='right_hip'),
12:
dict(
name='left_knee',
id=12,
color=[0, 255, 0],
type='lower',
swap='right_knee'),
13:
dict(
name='left_ankle',
id=13,
color=[0, 255, 0],
type='lower',
swap='right_ankle'),
14:
dict(name='root', id=14, color=[51, 153, 255], type='lower', swap=''),
15:
dict(name='spine', id=15, color=[51, 153, 255], type='upper', swap=''),
16:
dict(name='head', id=16, color=[51, 153, 255], type='upper', swap='')
},
skeleton_info={
0: dict(link=('neck', 'right_shoulder'), id=0, color=[255, 128, 0]),
1: dict(
link=('right_shoulder', 'right_elbow'), id=1, color=[255, 128, 0]),
2:
dict(link=('right_elbow', 'right_wrist'), id=2, color=[255, 128, 0]),
3: dict(link=('neck', 'left_shoulder'), id=3, color=[0, 255, 0]),
4: dict(link=('left_shoulder', 'left_elbow'), id=4, color=[0, 255, 0]),
5: dict(link=('left_elbow', 'left_wrist'), id=5, color=[0, 255, 0]),
6: dict(link=('root', 'right_hip'), id=6, color=[255, 128, 0]),
7: dict(link=('right_hip', 'right_knee'), id=7, color=[255, 128, 0]),
8: dict(link=('right_knee', 'right_ankle'), id=8, color=[255, 128, 0]),
9: dict(link=('root', 'left_hip'), id=9, color=[0, 255, 0]),
10: dict(link=('left_hip', 'left_knee'), id=10, color=[0, 255, 0]),
11: dict(link=('left_knee', 'left_ankle'), id=11, color=[0, 255, 0]),
12: dict(link=('head_top', 'head'), id=12, color=[51, 153, 255]),
13: dict(link=('head', 'neck'), id=13, color=[51, 153, 255]),
14: dict(link=('neck', 'spine'), id=14, color=[51, 153, 255]),
15: dict(link=('spine', 'root'), id=15, color=[51, 153, 255])
},
joint_weights=[1.] * 17,
sigmas=[])
configs/_base_/datasets/mpii.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='mpii',
paper_info=dict(
author='Mykhaylo Andriluka and Leonid Pishchulin and '
'Peter Gehler and Schiele, Bernt',
title='2D Human Pose Estimation: New Benchmark and '
'State of the Art Analysis',
container='IEEE Conference on Computer Vision and '
'Pattern Recognition (CVPR)',
year='2014',
homepage='http://human-pose.mpi-inf.mpg.de/',
),
keypoint_info={
0:
dict(
name='right_ankle',
id=0,
color=[255, 128, 0],
type='lower',
swap='left_ankle'),
1:
dict(
name='right_knee',
id=1,
color=[255, 128, 0],
type='lower',
swap='left_knee'),
2:
dict(
name='right_hip',
id=2,
color=[255, 128, 0],
type='lower',
swap='left_hip'),
3:
dict(
name='left_hip',
id=3,
color=[0, 255, 0],
type='lower',
swap='right_hip'),
4:
dict(
name='left_knee',
id=4,
color=[0, 255, 0],
type='lower',
swap='right_knee'),
5:
dict(
name='left_ankle',
id=5,
color=[0, 255, 0],
type='lower',
swap='right_ankle'),
6:
dict(name='pelvis', id=6, color=[51, 153, 255], type='lower', swap=''),
7:
dict(name='thorax', id=7, color=[51, 153, 255], type='upper', swap=''),
8:
dict(
name='upper_neck',
id=8,
color=[51, 153, 255],
type='upper',
swap=''),
9:
dict(
name='head_top', id=9, color=[51, 153, 255], type='upper',
swap=''),
10:
dict(
name='right_wrist',
id=10,
color=[255, 128, 0],
type='upper',
swap='left_wrist'),
11:
dict(
name='right_elbow',
id=11,
color=[255, 128, 0],
type='upper',
swap='left_elbow'),
12:
dict(
name='right_shoulder',
id=12,
color=[255, 128, 0],
type='upper',
swap='left_shoulder'),
13:
dict(
name='left_shoulder',
id=13,
color=[0, 255, 0],
type='upper',
swap='right_shoulder'),
14:
dict(
name='left_elbow',
id=14,
color=[0, 255, 0],
type='upper',
swap='right_elbow'),
15:
dict(
name='left_wrist',
id=15,
color=[0, 255, 0],
type='upper',
swap='right_wrist')
},
skeleton_info={
0:
dict(link=('right_ankle', 'right_knee'), id=0, color=[255, 128, 0]),
1:
dict(link=('right_knee', 'right_hip'), id=1, color=[255, 128, 0]),
2:
dict(link=('right_hip', 'pelvis'), id=2, color=[255, 128, 0]),
3:
dict(link=('pelvis', 'left_hip'), id=3, color=[0, 255, 0]),
4:
dict(link=('left_hip', 'left_knee'), id=4, color=[0, 255, 0]),
5:
dict(link=('left_knee', 'left_ankle'), id=5, color=[0, 255, 0]),
6:
dict(link=('pelvis', 'thorax'), id=6, color=[51, 153, 255]),
7:
dict(link=('thorax', 'upper_neck'), id=7, color=[51, 153, 255]),
8:
dict(link=('upper_neck', 'head_top'), id=8, color=[51, 153, 255]),
9:
dict(link=('upper_neck', 'right_shoulder'), id=9, color=[255, 128, 0]),
10:
dict(
link=('right_shoulder', 'right_elbow'), id=10, color=[255, 128,
0]),
11:
dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
12:
dict(link=('upper_neck', 'left_shoulder'), id=12, color=[0, 255, 0]),
13:
dict(link=('left_shoulder', 'left_elbow'), id=13, color=[0, 255, 0]),
14:
dict(link=('left_elbow', 'left_wrist'), id=14, color=[0, 255, 0])
},
joint_weights=[
1.5, 1.2, 1., 1., 1.2, 1.5, 1., 1., 1., 1., 1.5, 1.2, 1., 1., 1.2, 1.5
],
# Adapted from COCO dataset.
sigmas=[
0.089, 0.083, 0.107, 0.107, 0.083, 0.089, 0.026, 0.026, 0.026, 0.026,
0.062, 0.072, 0.179, 0.179, 0.072, 0.062
])
configs/_base_/datasets/mpii_trb.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='mpii_trb',
paper_info=dict(
author='Duan, Haodong and Lin, Kwan-Yee and Jin, Sheng and '
'Liu, Wentao and Qian, Chen and Ouyang, Wanli',
title='TRB: A Novel Triplet Representation for '
'Understanding 2D Human Body',
container='Proceedings of the IEEE International '
'Conference on Computer Vision',
year='2019',
homepage='https://github.com/kennymckormick/'
'Triplet-Representation-of-human-Body',
),
keypoint_info={
0:
dict(
name='left_shoulder',
id=0,
color=[0, 255, 0],
type='upper',
swap='right_shoulder'),
1:
dict(
name='right_shoulder',
id=1,
color=[255, 128, 0],
type='upper',
swap='left_shoulder'),
2:
dict(
name='left_elbow',
id=2,
color=[0, 255, 0],
type='upper',
swap='right_elbow'),
3:
dict(
name='right_elbow',
id=3,
color=[255, 128, 0],
type='upper',
swap='left_elbow'),
4:
dict(
name='left_wrist',
id=4,
color=[0, 255, 0],
type='upper',
swap='right_wrist'),
5:
dict(
name='right_wrist',
id=5,
color=[255, 128, 0],
type='upper',
swap='left_wrist'),
6:
dict(
name='left_hip',
id=6,
color=[0, 255, 0],
type='lower',
swap='right_hip'),
7:
dict(
name='right_hip',
id=7,
color=[255, 128, 0],
type='lower',
swap='left_hip'),
8:
dict(
name='left_knee',
id=8,
color=[0, 255, 0],
type='lower',
swap='right_knee'),
9:
dict(
name='right_knee',
id=9,
color=[255, 128, 0],
type='lower',
swap='left_knee'),
10:
dict(
name='left_ankle',
id=10,
color=[0, 255, 0],
type='lower',
swap='right_ankle'),
11:
dict(
name='right_ankle',
id=11,
color=[255, 128, 0],
type='lower',
swap='left_ankle'),
12:
dict(name='head', id=12, color=[51, 153, 255], type='upper', swap=''),
13:
dict(name='neck', id=13, color=[51, 153, 255], type='upper', swap=''),
14:
dict(
name='right_neck',
id=14,
color=[255, 255, 255],
type='upper',
swap='left_neck'),
15:
dict(
name='left_neck',
id=15,
color=[255, 255, 255],
type='upper',
swap='right_neck'),
16:
dict(
name='medial_right_shoulder',
id=16,
color=[255, 255, 255],
type='upper',
swap='medial_left_shoulder'),
17:
dict(
name='lateral_right_shoulder',
id=17,
color=[255, 255, 255],
type='upper',
swap='lateral_left_shoulder'),
18:
dict(
name='medial_right_bow',
id=18,
color=[255, 255, 255],
type='upper',
swap='medial_left_bow'),
19:
dict(
name='lateral_right_bow',
id=19,
color=[255, 255, 255],
type='upper',
swap='lateral_left_bow'),
20:
dict(
name='medial_right_wrist',
id=20,
color=[255, 255, 255],
type='upper',
swap='medial_left_wrist'),
21:
dict(
name='lateral_right_wrist',
id=21,
color=[255, 255, 255],
type='upper',
swap='lateral_left_wrist'),
22:
dict(
name='medial_left_shoulder',
id=22,
color=[255, 255, 255],
type='upper',
swap='medial_right_shoulder'),
23:
dict(
name='lateral_left_shoulder',
id=23,
color=[255, 255, 255],
type='upper',
swap='lateral_right_shoulder'),
24:
dict(
name='medial_left_bow',
id=24,
color=[255, 255, 255],
type='upper',
swap='medial_right_bow'),
25:
dict(
name='lateral_left_bow',
id=25,
color=[255, 255, 255],
type='upper',
swap='lateral_right_bow'),
26:
dict(
name='medial_left_wrist',
id=26,
color=[255, 255, 255],
type='upper',
swap='medial_right_wrist'),
27:
dict(
name='lateral_left_wrist',
id=27,
color=[255, 255, 255],
type='upper',
swap='lateral_right_wrist'),
28:
dict(
name='medial_right_hip',
id=28,
color=[255, 255, 255],
type='lower',
swap='medial_left_hip'),
29:
dict(
name='lateral_right_hip',
id=29,
color=[255, 255, 255],
type='lower',
swap='lateral_left_hip'),
30:
dict(
name='medial_right_knee',
id=30,
color=[255, 255, 255],
type='lower',
swap='medial_left_knee'),
31:
dict(
name='lateral_right_knee',
id=31,
color=[255, 255, 255],
type='lower',
swap='lateral_left_knee'),
32:
dict(
name='medial_right_ankle',
id=32,
color=[255, 255, 255],
type='lower',
swap='medial_left_ankle'),
33:
dict(
name='lateral_right_ankle',
id=33,
color=[255, 255, 255],
type='lower',
swap='lateral_left_ankle'),
34:
dict(
name='medial_left_hip',
id=34,
color=[255, 255, 255],
type='lower',
swap='medial_right_hip'),
35:
dict(
name='lateral_left_hip',
id=35,
color=[255, 255, 255],
type='lower',
swap='lateral_right_hip'),
36:
dict(
name='medial_left_knee',
id=36,
color=[255, 255, 255],
type='lower',
swap='medial_right_knee'),
37:
dict(
name='lateral_left_knee',
id=37,
color=[255, 255, 255],
type='lower',
swap='lateral_right_knee'),
38:
dict(
name='medial_left_ankle',
id=38,
color=[255, 255, 255],
type='lower',
swap='medial_right_ankle'),
39:
dict(
name='lateral_left_ankle',
id=39,
color=[255, 255, 255],
type='lower',
swap='lateral_right_ankle'),
},
skeleton_info={
0:
dict(link=('head', 'neck'), id=0, color=[51, 153, 255]),
1:
dict(link=('neck', 'left_shoulder'), id=1, color=[51, 153, 255]),
2:
dict(link=('neck', 'right_shoulder'), id=2, color=[51, 153, 255]),
3:
dict(link=('left_shoulder', 'left_elbow'), id=3, color=[0, 255, 0]),
4:
dict(
link=('right_shoulder', 'right_elbow'), id=4, color=[255, 128, 0]),
5:
dict(link=('left_elbow', 'left_wrist'), id=5, color=[0, 255, 0]),
6:
dict(link=('right_elbow', 'right_wrist'), id=6, color=[255, 128, 0]),
7:
dict(link=('left_shoulder', 'left_hip'), id=7, color=[51, 153, 255]),
8:
dict(link=('right_shoulder', 'right_hip'), id=8, color=[51, 153, 255]),
9:
dict(link=('left_hip', 'right_hip'), id=9, color=[51, 153, 255]),
10:
dict(link=('left_hip', 'left_knee'), id=10, color=[0, 255, 0]),
11:
dict(link=('right_hip', 'right_knee'), id=11, color=[255, 128, 0]),
12:
dict(link=('left_knee', 'left_ankle'), id=12, color=[0, 255, 0]),
13:
dict(link=('right_knee', 'right_ankle'), id=13, color=[255, 128, 0]),
14:
dict(link=('right_neck', 'left_neck'), id=14, color=[255, 255, 255]),
15:
dict(
link=('medial_right_shoulder', 'lateral_right_shoulder'),
id=15,
color=[255, 255, 255]),
16:
dict(
link=('medial_right_bow', 'lateral_right_bow'),
id=16,
color=[255, 255, 255]),
17:
dict(
link=('medial_right_wrist', 'lateral_right_wrist'),
id=17,
color=[255, 255, 255]),
18:
dict(
link=('medial_left_shoulder', 'lateral_left_shoulder'),
id=18,
color=[255, 255, 255]),
19:
dict(
link=('medial_left_bow', 'lateral_left_bow'),
id=19,
color=[255, 255, 255]),
20:
dict(
link=('medial_left_wrist', 'lateral_left_wrist'),
id=20,
color=[255, 255, 255]),
21:
dict(
link=('medial_right_hip', 'lateral_right_hip'),
id=21,
color=[255, 255, 255]),
22:
dict(
link=('medial_right_knee', 'lateral_right_knee'),
id=22,
color=[255, 255, 255]),
23:
dict(
link=('medial_right_ankle', 'lateral_right_ankle'),
id=23,
color=[255, 255, 255]),
24:
dict(
link=('medial_left_hip', 'lateral_left_hip'),
id=24,
color=[255, 255, 255]),
25:
dict(
link=('medial_left_knee', 'lateral_left_knee'),
id=25,
color=[255, 255, 255]),
26:
dict(
link=('medial_left_ankle', 'lateral_left_ankle'),
id=26,
color=[255, 255, 255])
},
joint_weights=[1.] * 40,
sigmas=[])
configs/_base_/datasets/ochuman.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='ochuman',
paper_info=dict(
author='Zhang, Song-Hai and Li, Ruilong and Dong, Xin and '
'Rosin, Paul and Cai, Zixi and Han, Xi and '
'Yang, Dingcheng and Huang, Haozhi and Hu, Shi-Min',
title='Pose2seg: Detection free human instance segmentation',
container='Proceedings of the IEEE conference on computer '
'vision and pattern recognition',
year='2019',
homepage='https://github.com/liruilong940607/OCHumanApi',
),
keypoint_info={
0:
dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
1:
dict(
name='left_eye',
id=1,
color=[51, 153, 255],
type='upper',
swap='right_eye'),
2:
dict(
name='right_eye',
id=2,
color=[51, 153, 255],
type='upper',
swap='left_eye'),
3:
dict(
name='left_ear',
id=3,
color=[51, 153, 255],
type='upper',
swap='right_ear'),
4:
dict(
name='right_ear',
id=4,
color=[51, 153, 255],
type='upper',
swap='left_ear'),
5:
dict(
name='left_shoulder',
id=5,
color=[0, 255, 0],
type='upper',
swap='right_shoulder'),
6:
dict(
name='right_shoulder',
id=6,
color=[255, 128, 0],
type='upper',
swap='left_shoulder'),
7:
dict(
name='left_elbow',
id=7,
color=[0, 255, 0],
type='upper',
swap='right_elbow'),
8:
dict(
name='right_elbow',
id=8,
color=[255, 128, 0],
type='upper',
swap='left_elbow'),
9:
dict(
name='left_wrist',
id=9,
color=[0, 255, 0],
type='upper',
swap='right_wrist'),
10:
dict(
name='right_wrist',
id=10,
color=[255, 128, 0],
type='upper',
swap='left_wrist'),
11:
dict(
name='left_hip',
id=11,
color=[0, 255, 0],
type='lower',
swap='right_hip'),
12:
dict(
name='right_hip',
id=12,
color=[255, 128, 0],
type='lower',
swap='left_hip'),
13:
dict(
name='left_knee',
id=13,
color=[0, 255, 0],
type='lower',
swap='right_knee'),
14:
dict(
name='right_knee',
id=14,
color=[255, 128, 0],
type='lower',
swap='left_knee'),
15:
dict(
name='left_ankle',
id=15,
color=[0, 255, 0],
type='lower',
swap='right_ankle'),
16:
dict(
name='right_ankle',
id=16,
color=[255, 128, 0],
type='lower',
swap='left_ankle')
},
skeleton_info={
0:
dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
1:
dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
2:
dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
3:
dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
4:
dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
5:
dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
6:
dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
7:
dict(
link=('left_shoulder', 'right_shoulder'),
id=7,
color=[51, 153, 255]),
8:
dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
9:
dict(
link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
10:
dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
11:
dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
12:
dict(link=('left_eye', 'right_eye'), id=12, color=[51, 153, 255]),
13:
dict(link=('nose', 'left_eye'), id=13, color=[51, 153, 255]),
14:
dict(link=('nose', 'right_eye'), id=14, color=[51, 153, 255]),
15:
dict(link=('left_eye', 'left_ear'), id=15, color=[51, 153, 255]),
16:
dict(link=('right_eye', 'right_ear'), id=16, color=[51, 153, 255]),
17:
dict(link=('left_ear', 'left_shoulder'), id=17, color=[51, 153, 255]),
18:
dict(
link=('right_ear', 'right_shoulder'), id=18, color=[51, 153, 255])
},
joint_weights=[
1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.5, 1.5, 1., 1., 1.2, 1.2, 1.5,
1.5
],
sigmas=[
0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089
])
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment