File size: 7,793 Bytes
2cd560a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
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=[])