Skip to content

Commit 2016609

Browse files
authored
[Feature] Add inferencers (#1969)
1 parent 4e62b72 commit 2016609

File tree

28 files changed

+2272
-1683
lines changed

28 files changed

+2272
-1683
lines changed

configs/_base_/datasets/300w.py

Lines changed: 91 additions & 341 deletions
Large diffs are not rendered by default.

configs/_base_/datasets/aflw.py

Lines changed: 19 additions & 58 deletions
Original file line numberDiff line numberDiff line change
@@ -13,70 +13,31 @@
1313
'team-bischof/lrs/downloads/aflw/',
1414
),
1515
keypoint_info={
16-
0:
17-
dict(name='kpt-0', id=0, color=[255, 255, 255], type='', swap='kpt-5'),
18-
1:
19-
dict(name='kpt-1', id=1, color=[255, 255, 255], type='', swap='kpt-4'),
20-
2:
21-
dict(name='kpt-2', id=2, color=[255, 255, 255], type='', swap='kpt-3'),
22-
3:
23-
dict(name='kpt-3', id=3, color=[255, 255, 255], type='', swap='kpt-2'),
24-
4:
25-
dict(name='kpt-4', id=4, color=[255, 255, 255], type='', swap='kpt-1'),
26-
5:
27-
dict(name='kpt-5', id=5, color=[255, 255, 255], type='', swap='kpt-0'),
28-
6:
29-
dict(
30-
name='kpt-6', id=6, color=[255, 255, 255], type='', swap='kpt-11'),
31-
7:
32-
dict(
33-
name='kpt-7', id=7, color=[255, 255, 255], type='', swap='kpt-10'),
34-
8:
35-
dict(name='kpt-8', id=8, color=[255, 255, 255], type='', swap='kpt-9'),
36-
9:
37-
dict(name='kpt-9', id=9, color=[255, 255, 255], type='', swap='kpt-8'),
16+
0: dict(name='kpt-0', id=0, color=[255, 0, 0], type='', swap='kpt-5'),
17+
1: dict(name='kpt-1', id=1, color=[255, 0, 0], type='', swap='kpt-4'),
18+
2: dict(name='kpt-2', id=2, color=[255, 0, 0], type='', swap='kpt-3'),
19+
3: dict(name='kpt-3', id=3, color=[255, 0, 0], type='', swap='kpt-2'),
20+
4: dict(name='kpt-4', id=4, color=[255, 0, 0], type='', swap='kpt-1'),
21+
5: dict(name='kpt-5', id=5, color=[255, 0, 0], type='', swap='kpt-0'),
22+
6: dict(name='kpt-6', id=6, color=[255, 0, 0], type='', swap='kpt-11'),
23+
7: dict(name='kpt-7', id=7, color=[255, 0, 0], type='', swap='kpt-10'),
24+
8: dict(name='kpt-8', id=8, color=[255, 0, 0], type='', swap='kpt-9'),
25+
9: dict(name='kpt-9', id=9, color=[255, 0, 0], type='', swap='kpt-8'),
3826
10:
39-
dict(
40-
name='kpt-10', id=10, color=[255, 255, 255], type='',
41-
swap='kpt-7'),
27+
dict(name='kpt-10', id=10, color=[255, 0, 0], type='', swap='kpt-7'),
4228
11:
43-
dict(
44-
name='kpt-11', id=11, color=[255, 255, 255], type='',
45-
swap='kpt-6'),
29+
dict(name='kpt-11', id=11, color=[255, 0, 0], type='', swap='kpt-6'),
4630
12:
47-
dict(
48-
name='kpt-12',
49-
id=12,
50-
color=[255, 255, 255],
51-
type='',
52-
swap='kpt-14'),
53-
13:
54-
dict(name='kpt-13', id=13, color=[255, 255, 255], type='', swap=''),
31+
dict(name='kpt-12', id=12, color=[255, 0, 0], type='', swap='kpt-14'),
32+
13: dict(name='kpt-13', id=13, color=[255, 0, 0], type='', swap=''),
5533
14:
56-
dict(
57-
name='kpt-14',
58-
id=14,
59-
color=[255, 255, 255],
60-
type='',
61-
swap='kpt-12'),
34+
dict(name='kpt-14', id=14, color=[255, 0, 0], type='', swap='kpt-12'),
6235
15:
63-
dict(
64-
name='kpt-15',
65-
id=15,
66-
color=[255, 255, 255],
67-
type='',
68-
swap='kpt-17'),
69-
16:
70-
dict(name='kpt-16', id=16, color=[255, 255, 255], type='', swap=''),
36+
dict(name='kpt-15', id=15, color=[255, 0, 0], type='', swap='kpt-17'),
37+
16: dict(name='kpt-16', id=16, color=[255, 0, 0], type='', swap=''),
7138
17:
72-
dict(
73-
name='kpt-17',
74-
id=17,
75-
color=[255, 255, 255],
76-
type='',
77-
swap='kpt-15'),
78-
18:
79-
dict(name='kpt-18', id=18, color=[255, 255, 255], type='', swap='')
39+
dict(name='kpt-17', id=17, color=[255, 0, 0], type='', swap='kpt-15'),
40+
18: dict(name='kpt-18', id=18, color=[255, 0, 0], type='', swap='')
8041
},
8142
skeleton_info={},
8243
joint_weights=[1.] * 19,

0 commit comments

Comments
 (0)