You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
2.7 KiB
77 lines
2.7 KiB
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from mmpose.datasets import DatasetInfo
|
|
|
|
|
|
def test_dataset_info():
|
|
dataset_info = dict(
|
|
dataset_name='zebra',
|
|
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='snout', id=0, color=[255, 255, 255], type='', swap=''),
|
|
1:
|
|
dict(name='head', id=1, color=[255, 255, 255], type='', swap=''),
|
|
2:
|
|
dict(name='neck', id=2, color=[255, 255, 255], type='', swap=''),
|
|
3:
|
|
dict(
|
|
name='forelegL1',
|
|
id=3,
|
|
color=[255, 255, 255],
|
|
type='',
|
|
swap='forelegR1'),
|
|
4:
|
|
dict(
|
|
name='forelegR1',
|
|
id=4,
|
|
color=[255, 255, 255],
|
|
type='',
|
|
swap='forelegL1'),
|
|
5:
|
|
dict(
|
|
name='hindlegL1',
|
|
id=5,
|
|
color=[255, 255, 255],
|
|
type='',
|
|
swap='hindlegR1'),
|
|
6:
|
|
dict(
|
|
name='hindlegR1',
|
|
id=6,
|
|
color=[255, 255, 255],
|
|
type='',
|
|
swap='hindlegL1'),
|
|
7:
|
|
dict(
|
|
name='tailbase', id=7, color=[255, 255, 255], type='',
|
|
swap=''),
|
|
8:
|
|
dict(
|
|
name='tailtip', id=8, color=[255, 255, 255], type='', swap='')
|
|
},
|
|
skeleton_info={
|
|
0: dict(link=('head', 'snout'), id=0, color=[255, 255, 255]),
|
|
1: dict(link=('neck', 'head'), id=1, color=[255, 255, 255]),
|
|
2: dict(link=('forelegL1', 'neck'), id=2, color=[255, 255, 255]),
|
|
3: dict(link=('forelegR1', 'neck'), id=3, color=[255, 255, 255]),
|
|
4:
|
|
dict(link=('hindlegL1', 'tailbase'), id=4, color=[255, 255, 255]),
|
|
5:
|
|
dict(link=('hindlegR1', 'tailbase'), id=5, color=[255, 255, 255]),
|
|
6: dict(link=('tailbase', 'neck'), id=6, color=[255, 255, 255]),
|
|
7: dict(link=('tailtip', 'tailbase'), id=7, color=[255, 255, 255])
|
|
},
|
|
joint_weights=[1.] * 9,
|
|
sigmas=[])
|
|
|
|
dataset_info = DatasetInfo(dataset_info)
|
|
assert dataset_info.keypoint_num == len(dataset_info.flip_index)
|
|
|