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
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 2223 additions and 0 deletions
configs/_base_/datasets/ak.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='Animal Kingdom',
paper_info=dict(
author='Singapore University of Technology and Design, Singapore.'
' Xun Long Ng, Kian Eng Ong, Qichen Zheng,'
' Yun Ni, Si Yong Yeo, Jun Liu.',
title='Animal Kingdom: '
'A Large and Diverse Dataset for Animal Behavior Understanding',
container='Conference on Computer Vision '
'and Pattern Recognition (CVPR)',
year='2022',
homepage='https://sutdcv.github.io/Animal-Kingdom',
version='1.0 (2022-06)',
date_created='2022-06',
),
keypoint_info={
0:
dict(
name='Head_Mid_Top',
id=0,
color=(225, 0, 255),
type='upper',
swap=''),
1:
dict(
name='Eye_Left',
id=1,
color=[220, 20, 60],
type='upper',
swap='Eye_Right'),
2:
dict(
name='Eye_Right',
id=2,
color=[0, 255, 255],
type='upper',
swap='Eye_Left'),
3:
dict(
name='Mouth_Front_Top',
id=3,
color=(0, 255, 42),
type='upper',
swap=''),
4:
dict(
name='Mouth_Back_Left',
id=4,
color=[221, 160, 221],
type='upper',
swap='Mouth_Back_Right'),
5:
dict(
name='Mouth_Back_Right',
id=5,
color=[135, 206, 250],
type='upper',
swap='Mouth_Back_Left'),
6:
dict(
name='Mouth_Front_Bottom',
id=6,
color=[50, 205, 50],
type='upper',
swap=''),
7:
dict(
name='Shoulder_Left',
id=7,
color=[255, 182, 193],
type='upper',
swap='Shoulder_Right'),
8:
dict(
name='Shoulder_Right',
id=8,
color=[0, 191, 255],
type='upper',
swap='Shoulder_Left'),
9:
dict(
name='Elbow_Left',
id=9,
color=[255, 105, 180],
type='upper',
swap='Elbow_Right'),
10:
dict(
name='Elbow_Right',
id=10,
color=[30, 144, 255],
type='upper',
swap='Elbow_Left'),
11:
dict(
name='Wrist_Left',
id=11,
color=[255, 20, 147],
type='upper',
swap='Wrist_Right'),
12:
dict(
name='Wrist_Right',
id=12,
color=[0, 0, 255],
type='upper',
swap='Wrist_Left'),
13:
dict(
name='Torso_Mid_Back',
id=13,
color=(185, 3, 221),
type='upper',
swap=''),
14:
dict(
name='Hip_Left',
id=14,
color=[255, 215, 0],
type='lower',
swap='Hip_Right'),
15:
dict(
name='Hip_Right',
id=15,
color=[147, 112, 219],
type='lower',
swap='Hip_Left'),
16:
dict(
name='Knee_Left',
id=16,
color=[255, 165, 0],
type='lower',
swap='Knee_Right'),
17:
dict(
name='Knee_Right',
id=17,
color=[138, 43, 226],
type='lower',
swap='Knee_Left'),
18:
dict(
name='Ankle_Left',
id=18,
color=[255, 140, 0],
type='lower',
swap='Ankle_Right'),
19:
dict(
name='Ankle_Right',
id=19,
color=[128, 0, 128],
type='lower',
swap='Ankle_Left'),
20:
dict(
name='Tail_Top_Back',
id=20,
color=(0, 251, 255),
type='lower',
swap=''),
21:
dict(
name='Tail_Mid_Back',
id=21,
color=[32, 178, 170],
type='lower',
swap=''),
22:
dict(
name='Tail_End_Back',
id=22,
color=(0, 102, 102),
type='lower',
swap='')
},
skeleton_info={
0:
dict(link=('Eye_Left', 'Head_Mid_Top'), id=0, color=[220, 20, 60]),
1:
dict(link=('Eye_Right', 'Head_Mid_Top'), id=1, color=[0, 255, 255]),
2:
dict(
link=('Mouth_Front_Top', 'Mouth_Back_Left'),
id=2,
color=[221, 160, 221]),
3:
dict(
link=('Mouth_Front_Top', 'Mouth_Back_Right'),
id=3,
color=[135, 206, 250]),
4:
dict(
link=('Mouth_Front_Bottom', 'Mouth_Back_Left'),
id=4,
color=[221, 160, 221]),
5:
dict(
link=('Mouth_Front_Bottom', 'Mouth_Back_Right'),
id=5,
color=[135, 206, 250]),
6:
dict(
link=('Head_Mid_Top', 'Torso_Mid_Back'), id=6,
color=(225, 0, 255)),
7:
dict(
link=('Torso_Mid_Back', 'Tail_Top_Back'),
id=7,
color=(185, 3, 221)),
8:
dict(
link=('Tail_Top_Back', 'Tail_Mid_Back'), id=8,
color=(0, 251, 255)),
9:
dict(
link=('Tail_Mid_Back', 'Tail_End_Back'),
id=9,
color=[32, 178, 170]),
10:
dict(
link=('Head_Mid_Top', 'Shoulder_Left'),
id=10,
color=[255, 182, 193]),
11:
dict(
link=('Head_Mid_Top', 'Shoulder_Right'),
id=11,
color=[0, 191, 255]),
12:
dict(
link=('Shoulder_Left', 'Elbow_Left'), id=12, color=[255, 105,
180]),
13:
dict(
link=('Shoulder_Right', 'Elbow_Right'),
id=13,
color=[30, 144, 255]),
14:
dict(link=('Elbow_Left', 'Wrist_Left'), id=14, color=[255, 20, 147]),
15:
dict(link=('Elbow_Right', 'Wrist_Right'), id=15, color=[0, 0, 255]),
16:
dict(link=('Tail_Top_Back', 'Hip_Left'), id=16, color=[255, 215, 0]),
17:
dict(
link=('Tail_Top_Back', 'Hip_Right'), id=17, color=[147, 112, 219]),
18:
dict(link=('Hip_Left', 'Knee_Left'), id=18, color=[255, 165, 0]),
19:
dict(link=('Hip_Right', 'Knee_Right'), id=19, color=[138, 43, 226]),
20:
dict(link=('Knee_Left', 'Ankle_Left'), id=20, color=[255, 140, 0]),
21:
dict(link=('Knee_Right', 'Ankle_Right'), id=21, color=[128, 0, 128])
},
joint_weights=[
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
1., 1., 1., 1., 1.
],
sigmas=[
0.025, 0.025, 0.025, 0.025, 0.025, 0.025, 0.025, 0.025, 0.025, 0.025,
0.025, 0.025, 0.025, 0.025, 0.025, 0.025, 0.025, 0.025, 0.025, 0.025,
0.025, 0.025, 0.025
])
configs/_base_/datasets/animalpose.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='animalpose',
paper_info=dict(
author='Cao, Jinkun and Tang, Hongyang and Fang, Hao-Shu and '
'Shen, Xiaoyong and Lu, Cewu and Tai, Yu-Wing',
title='Cross-Domain Adaptation for Animal Pose Estimation',
container='The IEEE International Conference on '
'Computer Vision (ICCV)',
year='2019',
homepage='https://sites.google.com/view/animal-pose/',
),
keypoint_info={
0:
dict(
name='L_Eye', id=0, color=[0, 255, 0], type='upper', swap='R_Eye'),
1:
dict(
name='R_Eye',
id=1,
color=[255, 128, 0],
type='upper',
swap='L_Eye'),
2:
dict(
name='L_EarBase',
id=2,
color=[0, 255, 0],
type='upper',
swap='R_EarBase'),
3:
dict(
name='R_EarBase',
id=3,
color=[255, 128, 0],
type='upper',
swap='L_EarBase'),
4:
dict(name='Nose', id=4, color=[51, 153, 255], type='upper', swap=''),
5:
dict(name='Throat', id=5, color=[51, 153, 255], type='upper', swap=''),
6:
dict(
name='TailBase', id=6, color=[51, 153, 255], type='lower',
swap=''),
7:
dict(
name='Withers', id=7, color=[51, 153, 255], type='upper', swap=''),
8:
dict(
name='L_F_Elbow',
id=8,
color=[0, 255, 0],
type='upper',
swap='R_F_Elbow'),
9:
dict(
name='R_F_Elbow',
id=9,
color=[255, 128, 0],
type='upper',
swap='L_F_Elbow'),
10:
dict(
name='L_B_Elbow',
id=10,
color=[0, 255, 0],
type='lower',
swap='R_B_Elbow'),
11:
dict(
name='R_B_Elbow',
id=11,
color=[255, 128, 0],
type='lower',
swap='L_B_Elbow'),
12:
dict(
name='L_F_Knee',
id=12,
color=[0, 255, 0],
type='upper',
swap='R_F_Knee'),
13:
dict(
name='R_F_Knee',
id=13,
color=[255, 128, 0],
type='upper',
swap='L_F_Knee'),
14:
dict(
name='L_B_Knee',
id=14,
color=[0, 255, 0],
type='lower',
swap='R_B_Knee'),
15:
dict(
name='R_B_Knee',
id=15,
color=[255, 128, 0],
type='lower',
swap='L_B_Knee'),
16:
dict(
name='L_F_Paw',
id=16,
color=[0, 255, 0],
type='upper',
swap='R_F_Paw'),
17:
dict(
name='R_F_Paw',
id=17,
color=[255, 128, 0],
type='upper',
swap='L_F_Paw'),
18:
dict(
name='L_B_Paw',
id=18,
color=[0, 255, 0],
type='lower',
swap='R_B_Paw'),
19:
dict(
name='R_B_Paw',
id=19,
color=[255, 128, 0],
type='lower',
swap='L_B_Paw')
},
skeleton_info={
0: dict(link=('L_Eye', 'R_Eye'), id=0, color=[51, 153, 255]),
1: dict(link=('L_Eye', 'L_EarBase'), id=1, color=[0, 255, 0]),
2: dict(link=('R_Eye', 'R_EarBase'), id=2, color=[255, 128, 0]),
3: dict(link=('L_Eye', 'Nose'), id=3, color=[0, 255, 0]),
4: dict(link=('R_Eye', 'Nose'), id=4, color=[255, 128, 0]),
5: dict(link=('Nose', 'Throat'), id=5, color=[51, 153, 255]),
6: dict(link=('Throat', 'Withers'), id=6, color=[51, 153, 255]),
7: dict(link=('TailBase', 'Withers'), id=7, color=[51, 153, 255]),
8: dict(link=('Throat', 'L_F_Elbow'), id=8, color=[0, 255, 0]),
9: dict(link=('L_F_Elbow', 'L_F_Knee'), id=9, color=[0, 255, 0]),
10: dict(link=('L_F_Knee', 'L_F_Paw'), id=10, color=[0, 255, 0]),
11: dict(link=('Throat', 'R_F_Elbow'), id=11, color=[255, 128, 0]),
12: dict(link=('R_F_Elbow', 'R_F_Knee'), id=12, color=[255, 128, 0]),
13: dict(link=('R_F_Knee', 'R_F_Paw'), id=13, color=[255, 128, 0]),
14: dict(link=('TailBase', 'L_B_Elbow'), id=14, color=[0, 255, 0]),
15: dict(link=('L_B_Elbow', 'L_B_Knee'), id=15, color=[0, 255, 0]),
16: dict(link=('L_B_Knee', 'L_B_Paw'), id=16, color=[0, 255, 0]),
17: dict(link=('TailBase', 'R_B_Elbow'), id=17, color=[255, 128, 0]),
18: dict(link=('R_B_Elbow', 'R_B_Knee'), id=18, color=[255, 128, 0]),
19: dict(link=('R_B_Knee', 'R_B_Paw'), id=19, color=[255, 128, 0])
},
joint_weights=[
1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.2, 1.2,
1.5, 1.5, 1.5, 1.5
],
# Note: The original paper did not provide enough information about
# the sigmas. We modified from 'https://github.com/cocodataset/'
# 'cocoapi/blob/master/PythonAPI/pycocotools/cocoeval.py#L523'
sigmas=[
0.025, 0.025, 0.026, 0.035, 0.035, 0.10, 0.10, 0.10, 0.107, 0.107,
0.107, 0.107, 0.087, 0.087, 0.087, 0.087, 0.089, 0.089, 0.089, 0.089
])
configs/_base_/datasets/ap10k.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='ap10k',
paper_info=dict(
author='Yu, Hang and Xu, Yufei and Zhang, Jing and '
'Zhao, Wei and Guan, Ziyu and Tao, Dacheng',
title='AP-10K: A Benchmark for Animal Pose Estimation in the Wild',
container='35th Conference on Neural Information Processing Systems '
'(NeurIPS 2021) Track on Datasets and Bench-marks.',
year='2021',
homepage='https://github.com/AlexTheBad/AP-10K',
),
keypoint_info={
0:
dict(
name='L_Eye', id=0, color=[0, 255, 0], type='upper', swap='R_Eye'),
1:
dict(
name='R_Eye',
id=1,
color=[255, 128, 0],
type='upper',
swap='L_Eye'),
2:
dict(name='Nose', id=2, color=[51, 153, 255], type='upper', swap=''),
3:
dict(name='Neck', id=3, color=[51, 153, 255], type='upper', swap=''),
4:
dict(
name='Root of tail',
id=4,
color=[51, 153, 255],
type='lower',
swap=''),
5:
dict(
name='L_Shoulder',
id=5,
color=[51, 153, 255],
type='upper',
swap='R_Shoulder'),
6:
dict(
name='L_Elbow',
id=6,
color=[51, 153, 255],
type='upper',
swap='R_Elbow'),
7:
dict(
name='L_F_Paw',
id=7,
color=[0, 255, 0],
type='upper',
swap='R_F_Paw'),
8:
dict(
name='R_Shoulder',
id=8,
color=[0, 255, 0],
type='upper',
swap='L_Shoulder'),
9:
dict(
name='R_Elbow',
id=9,
color=[255, 128, 0],
type='upper',
swap='L_Elbow'),
10:
dict(
name='R_F_Paw',
id=10,
color=[0, 255, 0],
type='lower',
swap='L_F_Paw'),
11:
dict(
name='L_Hip',
id=11,
color=[255, 128, 0],
type='lower',
swap='R_Hip'),
12:
dict(
name='L_Knee',
id=12,
color=[255, 128, 0],
type='lower',
swap='R_Knee'),
13:
dict(
name='L_B_Paw',
id=13,
color=[0, 255, 0],
type='lower',
swap='R_B_Paw'),
14:
dict(
name='R_Hip', id=14, color=[0, 255, 0], type='lower',
swap='L_Hip'),
15:
dict(
name='R_Knee',
id=15,
color=[0, 255, 0],
type='lower',
swap='L_Knee'),
16:
dict(
name='R_B_Paw',
id=16,
color=[0, 255, 0],
type='lower',
swap='L_B_Paw'),
},
skeleton_info={
0: dict(link=('L_Eye', 'R_Eye'), id=0, color=[0, 0, 255]),
1: dict(link=('L_Eye', 'Nose'), id=1, color=[0, 0, 255]),
2: dict(link=('R_Eye', 'Nose'), id=2, color=[0, 0, 255]),
3: dict(link=('Nose', 'Neck'), id=3, color=[0, 255, 0]),
4: dict(link=('Neck', 'Root of tail'), id=4, color=[0, 255, 0]),
5: dict(link=('Neck', 'L_Shoulder'), id=5, color=[0, 255, 255]),
6: dict(link=('L_Shoulder', 'L_Elbow'), id=6, color=[0, 255, 255]),
7: dict(link=('L_Elbow', 'L_F_Paw'), id=6, color=[0, 255, 255]),
8: dict(link=('Neck', 'R_Shoulder'), id=7, color=[6, 156, 250]),
9: dict(link=('R_Shoulder', 'R_Elbow'), id=8, color=[6, 156, 250]),
10: dict(link=('R_Elbow', 'R_F_Paw'), id=9, color=[6, 156, 250]),
11: dict(link=('Root of tail', 'L_Hip'), id=10, color=[0, 255, 255]),
12: dict(link=('L_Hip', 'L_Knee'), id=11, color=[0, 255, 255]),
13: dict(link=('L_Knee', 'L_B_Paw'), id=12, color=[0, 255, 255]),
14: dict(link=('Root of tail', 'R_Hip'), id=13, color=[6, 156, 250]),
15: dict(link=('R_Hip', 'R_Knee'), id=14, color=[6, 156, 250]),
16: dict(link=('R_Knee', 'R_B_Paw'), id=15, color=[6, 156, 250]),
},
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.025, 0.025, 0.026, 0.035, 0.035, 0.079, 0.072, 0.062, 0.079, 0.072,
0.062, 0.107, 0.087, 0.089, 0.107, 0.087, 0.089
])
configs/_base_/datasets/atrw.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='atrw',
paper_info=dict(
author='Li, Shuyuan and Li, Jianguo and Tang, Hanlin '
'and Qian, Rui and Lin, Weiyao',
title='ATRW: A Benchmark for Amur Tiger '
'Re-identification in the Wild',
container='Proceedings of the 28th ACM '
'International Conference on Multimedia',
year='2020',
homepage='https://cvwc2019.github.io/challenge.html',
),
keypoint_info={
0:
dict(
name='left_ear',
id=0,
color=[51, 153, 255],
type='upper',
swap='right_ear'),
1:
dict(
name='right_ear',
id=1,
color=[51, 153, 255],
type='upper',
swap='left_ear'),
2:
dict(name='nose', id=2, color=[51, 153, 255], type='upper', swap=''),
3:
dict(
name='right_shoulder',
id=3,
color=[255, 128, 0],
type='upper',
swap='left_shoulder'),
4:
dict(
name='right_front_paw',
id=4,
color=[255, 128, 0],
type='upper',
swap='left_front_paw'),
5:
dict(
name='left_shoulder',
id=5,
color=[0, 255, 0],
type='upper',
swap='right_shoulder'),
6:
dict(
name='left_front_paw',
id=6,
color=[0, 255, 0],
type='upper',
swap='right_front_paw'),
7:
dict(
name='right_hip',
id=7,
color=[255, 128, 0],
type='lower',
swap='left_hip'),
8:
dict(
name='right_knee',
id=8,
color=[255, 128, 0],
type='lower',
swap='left_knee'),
9:
dict(
name='right_back_paw',
id=9,
color=[255, 128, 0],
type='lower',
swap='left_back_paw'),
10:
dict(
name='left_hip',
id=10,
color=[0, 255, 0],
type='lower',
swap='right_hip'),
11:
dict(
name='left_knee',
id=11,
color=[0, 255, 0],
type='lower',
swap='right_knee'),
12:
dict(
name='left_back_paw',
id=12,
color=[0, 255, 0],
type='lower',
swap='right_back_paw'),
13:
dict(name='tail', id=13, color=[51, 153, 255], type='lower', swap=''),
14:
dict(
name='center', id=14, color=[51, 153, 255], type='lower', swap=''),
},
skeleton_info={
0:
dict(link=('left_ear', 'nose'), id=0, color=[51, 153, 255]),
1:
dict(link=('right_ear', 'nose'), id=1, color=[51, 153, 255]),
2:
dict(link=('nose', 'center'), id=2, color=[51, 153, 255]),
3:
dict(
link=('left_shoulder', 'left_front_paw'), id=3, color=[0, 255, 0]),
4:
dict(link=('left_shoulder', 'center'), id=4, color=[0, 255, 0]),
5:
dict(
link=('right_shoulder', 'right_front_paw'),
id=5,
color=[255, 128, 0]),
6:
dict(link=('right_shoulder', 'center'), id=6, color=[255, 128, 0]),
7:
dict(link=('tail', 'center'), id=7, color=[51, 153, 255]),
8:
dict(link=('right_back_paw', 'right_knee'), id=8, color=[255, 128, 0]),
9:
dict(link=('right_knee', 'right_hip'), id=9, color=[255, 128, 0]),
10:
dict(link=('right_hip', 'tail'), id=10, color=[255, 128, 0]),
11:
dict(link=('left_back_paw', 'left_knee'), id=11, color=[0, 255, 0]),
12:
dict(link=('left_knee', 'left_hip'), id=12, color=[0, 255, 0]),
13:
dict(link=('left_hip', 'tail'), id=13, color=[0, 255, 0]),
},
joint_weights=[1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.],
sigmas=[
0.0277, 0.0823, 0.0831, 0.0202, 0.0716, 0.0263, 0.0646, 0.0302, 0.0440,
0.0316, 0.0333, 0.0547, 0.0263, 0.0683, 0.0539
])
configs/_base_/datasets/campus.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='campus',
paper_info=dict(
author='Belagiannis, Vasileios and Amin, Sikandar and Andriluka, '
'Mykhaylo and Schiele, Bernt and Navab, Nassir and Ilic, Slobodan',
title='3D Pictorial Structures for Multiple Human Pose Estimation',
container='IEEE Computer Society Conference on Computer Vision and '
'Pattern Recognition (CVPR)',
year='2014',
homepage='http://campar.in.tum.de/Chair/MultiHumanPose',
),
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='right_wrist',
id=6,
color=[255, 128, 0],
type='upper',
swap='left_wrist'),
7:
dict(
name='right_elbow',
id=7,
color=[255, 128, 0],
type='upper',
swap='left_elbow'),
8:
dict(
name='right_shoulder',
id=8,
color=[255, 128, 0],
type='upper',
swap='left_shoulder'),
9:
dict(
name='left_shoulder',
id=9,
color=[0, 255, 0],
type='upper',
swap='right_shoulder'),
10:
dict(
name='left_elbow',
id=10,
color=[0, 255, 0],
type='upper',
swap='right_elbow'),
11:
dict(
name='left_wrist',
id=11,
color=[0, 255, 0],
type='upper',
swap='right_wrist'),
12:
dict(
name='bottom_head',
id=12,
color=[51, 153, 255],
type='upper',
swap=''),
13:
dict(
name='top_head',
id=13,
color=[51, 153, 255],
type='upper',
swap=''),
},
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=('left_hip', 'left_knee'), id=2, color=[0, 255, 0]),
3:
dict(link=('left_knee', 'left_ankle'), id=3, color=[0, 255, 0]),
4:
dict(link=('right_hip', 'left_hip'), id=4, color=[51, 153, 255]),
5:
dict(link=('right_wrist', 'right_elbow'), id=5, color=[255, 128, 0]),
6:
dict(
link=('right_elbow', 'right_shoulder'), id=6, color=[255, 128, 0]),
7:
dict(link=('left_shoulder', 'left_elbow'), id=7, color=[0, 255, 0]),
8:
dict(link=('left_elbow', 'left_wrist'), id=8, color=[0, 255, 0]),
9:
dict(link=('right_hip', 'right_shoulder'), id=9, color=[255, 128, 0]),
10:
dict(link=('left_hip', 'left_shoulder'), id=10, color=[0, 255, 0]),
11:
dict(
link=('right_shoulder', 'bottom_head'), id=11, color=[255, 128,
0]),
12:
dict(link=('left_shoulder', 'bottom_head'), id=12, color=[0, 255, 0]),
13:
dict(link=('bottom_head', 'top_head'), id=13, color=[51, 153, 255]),
},
joint_weights=[
1.5, 1.2, 1.0, 1.0, 1.2, 1.5, 1.5, 1.2, 1.0, 1.0, 1.2, 1.5, 1.0, 1.0
],
sigmas=[
0.089, 0.087, 0.107, 0.107, 0.087, 0.089, 0.062, 0.072, 0.079, 0.079,
0.072, 0.062, 0.026, 0.026
])
configs/_base_/datasets/coco.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='coco',
paper_info=dict(
author='Lin, Tsung-Yi and Maire, Michael and '
'Belongie, Serge and Hays, James and '
'Perona, Pietro and Ramanan, Deva and '
r'Doll{\'a}r, Piotr and Zitnick, C Lawrence',
title='Microsoft coco: Common objects in context',
container='European conference on computer vision',
year='2014',
homepage='http://cocodataset.org/',
),
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/coco_aic.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='coco',
paper_info=[
dict(
author='Lin, Tsung-Yi and Maire, Michael and '
'Belongie, Serge and Hays, James and '
'Perona, Pietro and Ramanan, Deva and '
r'Doll{\'a}r, Piotr and Zitnick, C Lawrence',
title='Microsoft coco: Common objects in context',
container='European conference on computer vision',
year='2014',
homepage='http://cocodataset.org/',
),
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/coco_openpose.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='coco_openpose',
paper_info=dict(
author='Zhe, Cao and Tomas, Simon and '
'Shih-En, Wei and Yaser, Sheikh',
title='OpenPose: Realtime Multi-Person 2D Pose '
'Estimation using Part Affinity Fields',
container='IEEE Transactions on Pattern Analysis '
'and Machine Intelligence',
year='2019',
homepage='https://github.com/CMU-Perceptual-Computing-Lab/openpose/',
),
keypoint_info={
0:
dict(name='nose', id=0, color=[255, 0, 0], type='upper', swap=''),
1:
dict(name='neck', id=1, color=[255, 85, 0], type='upper', swap=''),
2:
dict(
name='right_shoulder',
id=2,
color=[255, 170, 0],
type='upper',
swap='left_shoulder'),
3:
dict(
name='right_elbow',
id=3,
color=[255, 255, 0],
type='upper',
swap='left_elbow'),
4:
dict(
name='right_wrist',
id=4,
color=[170, 255, 0],
type='upper',
swap='left_wrist'),
5:
dict(
name='left_shoulder',
id=5,
color=[85, 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, 85],
type='upper',
swap='right_wrist'),
8:
dict(
name='right_hip',
id=8,
color=[0, 255, 170],
type='lower',
swap='left_hip'),
9:
dict(
name='right_knee',
id=9,
color=[0, 255, 255],
type='lower',
swap='left_knee'),
10:
dict(
name='right_ankle',
id=10,
color=[0, 170, 255],
type='lower',
swap='left_ankle'),
11:
dict(
name='left_hip',
id=11,
color=[0, 85, 255],
type='lower',
swap='right_hip'),
12:
dict(
name='left_knee',
id=12,
color=[0, 0, 255],
type='lower',
swap='right_knee'),
13:
dict(
name='left_ankle',
id=13,
color=[85, 0, 255],
type='lower',
swap='right_ankle'),
14:
dict(
name='right_eye',
id=14,
color=[170, 0, 255],
type='upper',
swap='left_eye'),
15:
dict(
name='left_eye',
id=15,
color=[255, 0, 255],
type='upper',
swap='right_eye'),
16:
dict(
name='right_ear',
id=16,
color=[255, 0, 170],
type='upper',
swap='left_ear'),
17:
dict(
name='left_ear',
id=17,
color=[255, 0, 85],
type='upper',
swap='right_ear'),
},
skeleton_info={
0: dict(link=('neck', 'right_shoulder'), id=0, color=[255, 0, 0]),
1: dict(link=('neck', 'left_shoulder'), id=1, color=[255, 85, 0]),
2: dict(
link=('right_shoulder', 'right_elbow'), id=2, color=[255, 170, 0]),
3:
dict(link=('right_elbow', 'right_wrist'), id=3, color=[255, 255, 0]),
4:
dict(link=('left_shoulder', 'left_elbow'), id=4, color=[170, 255, 0]),
5: dict(link=('left_elbow', 'left_wrist'), id=5, color=[85, 255, 0]),
6: dict(link=('neck', 'right_hip'), id=6, color=[0, 255, 0]),
7: dict(link=('right_hip', 'right_knee'), id=7, color=[0, 255, 85]),
8: dict(link=('right_knee', 'right_ankle'), id=8, color=[0, 255, 170]),
9: dict(link=('neck', 'left_hip'), id=9, color=[0, 255, 225]),
10: dict(link=('left_hip', 'left_knee'), id=10, color=[0, 170, 255]),
11: dict(link=('left_knee', 'left_ankle'), id=11, color=[0, 85, 255]),
12: dict(link=('neck', 'nose'), id=12, color=[0, 0, 255]),
13: dict(link=('nose', 'right_eye'), id=13, color=[255, 0, 170]),
14: dict(link=('right_eye', 'right_ear'), id=14, color=[170, 0, 255]),
15: dict(link=('nose', 'left_eye'), id=15, color=[255, 0, 255]),
16: dict(link=('left_eye', 'left_ear'), id=16, color=[255, 0, 170]),
},
joint_weights=[1.] * 18,
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.082
])
configs/_base_/datasets/coco_wholebody.py 0 → 100644
View file @ ca8a762a
This diff is collapsed.
configs/_base_/datasets/coco_wholebody_face.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='coco_wholebody_face',
paper_info=dict(
author='Jin, Sheng and Xu, Lumin and Xu, Jin and '
'Wang, Can and Liu, Wentao and '
'Qian, Chen and Ouyang, Wanli and Luo, Ping',
title='Whole-Body Human Pose Estimation in the Wild',
container='Proceedings of the European '
'Conference on Computer Vision (ECCV)',
year='2020',
homepage='https://github.com/jin-s13/COCO-WholeBody/',
),
keypoint_info={
0:
dict(name='face-0', id=0, color=[255, 0, 0], type='', swap='face-16'),
1:
dict(name='face-1', id=1, color=[255, 0, 0], type='', swap='face-15'),
2:
dict(name='face-2', id=2, color=[255, 0, 0], type='', swap='face-14'),
3:
dict(name='face-3', id=3, color=[255, 0, 0], type='', swap='face-13'),
4:
dict(name='face-4', id=4, color=[255, 0, 0], type='', swap='face-12'),
5:
dict(name='face-5', id=5, color=[255, 0, 0], type='', swap='face-11'),
6:
dict(name='face-6', id=6, color=[255, 0, 0], type='', swap='face-10'),
7:
dict(name='face-7', id=7, color=[255, 0, 0], type='', swap='face-9'),
8: dict(name='face-8', id=8, color=[255, 0, 0], type='', swap=''),
9:
dict(name='face-9', id=9, color=[255, 0, 0], type='', swap='face-7'),
10:
dict(name='face-10', id=10, color=[255, 0, 0], type='', swap='face-6'),
11:
dict(name='face-11', id=11, color=[255, 0, 0], type='', swap='face-5'),
12:
dict(name='face-12', id=12, color=[255, 0, 0], type='', swap='face-4'),
13:
dict(name='face-13', id=13, color=[255, 0, 0], type='', swap='face-3'),
14:
dict(name='face-14', id=14, color=[255, 0, 0], type='', swap='face-2'),
15:
dict(name='face-15', id=15, color=[255, 0, 0], type='', swap='face-1'),
16:
dict(name='face-16', id=16, color=[255, 0, 0], type='', swap='face-0'),
17: dict(
name='face-17', id=17, color=[255, 0, 0], type='', swap='face-26'),
18: dict(
name='face-18', id=18, color=[255, 0, 0], type='', swap='face-25'),
19: dict(
name='face-19', id=19, color=[255, 0, 0], type='', swap='face-24'),
20: dict(
name='face-20', id=20, color=[255, 0, 0], type='', swap='face-23'),
21: dict(
name='face-21', id=21, color=[255, 0, 0], type='', swap='face-22'),
22: dict(
name='face-22', id=22, color=[255, 0, 0], type='', swap='face-21'),
23: dict(
name='face-23', id=23, color=[255, 0, 0], type='', swap='face-20'),
24: dict(
name='face-24', id=24, color=[255, 0, 0], type='', swap='face-19'),
25: dict(
name='face-25', id=25, color=[255, 0, 0], type='', swap='face-18'),
26: dict(
name='face-26', id=26, color=[255, 0, 0], type='', swap='face-17'),
27: dict(name='face-27', id=27, color=[255, 0, 0], type='', swap=''),
28: dict(name='face-28', id=28, color=[255, 0, 0], type='', swap=''),
29: dict(name='face-29', id=29, color=[255, 0, 0], type='', swap=''),
30: dict(name='face-30', id=30, color=[255, 0, 0], type='', swap=''),
31: dict(
name='face-31', id=31, color=[255, 0, 0], type='', swap='face-35'),
32: dict(
name='face-32', id=32, color=[255, 0, 0], type='', swap='face-34'),
33: dict(name='face-33', id=33, color=[255, 0, 0], type='', swap=''),
34: dict(
name='face-34', id=34, color=[255, 0, 0], type='', swap='face-32'),
35: dict(
name='face-35', id=35, color=[255, 0, 0], type='', swap='face-31'),
36: dict(
name='face-36', id=36, color=[255, 0, 0], type='', swap='face-45'),
37: dict(
name='face-37', id=37, color=[255, 0, 0], type='', swap='face-44'),
38: dict(
name='face-38', id=38, color=[255, 0, 0], type='', swap='face-43'),
39: dict(
name='face-39', id=39, color=[255, 0, 0], type='', swap='face-42'),
40: dict(
name='face-40', id=40, color=[255, 0, 0], type='', swap='face-47'),
41: dict(
name='face-41', id=41, color=[255, 0, 0], type='', swap='face-46'),
42: dict(
name='face-42', id=42, color=[255, 0, 0], type='', swap='face-39'),
43: dict(
name='face-43', id=43, color=[255, 0, 0], type='', swap='face-38'),
44: dict(
name='face-44', id=44, color=[255, 0, 0], type='', swap='face-37'),
45: dict(
name='face-45', id=45, color=[255, 0, 0], type='', swap='face-36'),
46: dict(
name='face-46', id=46, color=[255, 0, 0], type='', swap='face-41'),
47: dict(
name='face-47', id=47, color=[255, 0, 0], type='', swap='face-40'),
48: dict(
name='face-48', id=48, color=[255, 0, 0], type='', swap='face-54'),
49: dict(
name='face-49', id=49, color=[255, 0, 0], type='', swap='face-53'),
50: dict(
name='face-50', id=50, color=[255, 0, 0], type='', swap='face-52'),
51: dict(name='face-51', id=52, color=[255, 0, 0], type='', swap=''),
52: dict(
name='face-52', id=52, color=[255, 0, 0], type='', swap='face-50'),
53: dict(
name='face-53', id=53, color=[255, 0, 0], type='', swap='face-49'),
54: dict(
name='face-54', id=54, color=[255, 0, 0], type='', swap='face-48'),
55: dict(
name='face-55', id=55, color=[255, 0, 0], type='', swap='face-59'),
56: dict(
name='face-56', id=56, color=[255, 0, 0], type='', swap='face-58'),
57: dict(name='face-57', id=57, color=[255, 0, 0], type='', swap=''),
58: dict(
name='face-58', id=58, color=[255, 0, 0], type='', swap='face-56'),
59: dict(
name='face-59', id=59, color=[255, 0, 0], type='', swap='face-55'),
60: dict(
name='face-60', id=60, color=[255, 0, 0], type='', swap='face-64'),
61: dict(
name='face-61', id=61, color=[255, 0, 0], type='', swap='face-63'),
62: dict(name='face-62', id=62, color=[255, 0, 0], type='', swap=''),
63: dict(
name='face-63', id=63, color=[255, 0, 0], type='', swap='face-61'),
64: dict(
name='face-64', id=64, color=[255, 0, 0], type='', swap='face-60'),
65: dict(
name='face-65', id=65, color=[255, 0, 0], type='', swap='face-67'),
66: dict(name='face-66', id=66, color=[255, 0, 0], type='', swap=''),
67: dict(
name='face-67', id=67, color=[255, 0, 0], type='', swap='face-65')
},
skeleton_info={},
joint_weights=[1.] * 68,
# 'https://github.com/jin-s13/COCO-WholeBody/blob/master/'
# 'evaluation/myeval_wholebody.py#L177'
sigmas=[
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
])
configs/_base_/datasets/coco_wholebody_hand.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='coco_wholebody_hand',
paper_info=dict(
author='Jin, Sheng and Xu, Lumin and Xu, Jin and '
'Wang, Can and Liu, Wentao and '
'Qian, Chen and Ouyang, Wanli and Luo, Ping',
title='Whole-Body Human Pose Estimation in the Wild',
container='Proceedings of the European '
'Conference on Computer Vision (ECCV)',
year='2020',
homepage='https://github.com/jin-s13/COCO-WholeBody/',
),
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=[
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/coco_wholebody_openpose.py 0 → 100644
View file @ ca8a762a
This diff is collapsed.
configs/_base_/datasets/cofw.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='cofw',
paper_info=dict(
author='Burgos-Artizzu, Xavier P and Perona, '
r'Pietro and Doll{\'a}r, Piotr',
title='Robust face landmark estimation under occlusion',
container='Proceedings of the IEEE international '
'conference on computer vision',
year='2013',
homepage='http://www.vision.caltech.edu/xpburgos/ICCV13/',
),
keypoint_info={
0: dict(name='kpt-0', id=0, color=[255, 0, 0], type='', swap='kpt-1'),
1: dict(name='kpt-1', id=1, color=[255, 0, 0], type='', swap='kpt-0'),
2: dict(name='kpt-2', id=2, color=[255, 0, 0], type='', swap='kpt-3'),
3: dict(name='kpt-3', id=3, color=[255, 0, 0], type='', swap='kpt-2'),
4: dict(name='kpt-4', id=4, color=[255, 0, 0], type='', swap='kpt-6'),
5: dict(name='kpt-5', id=5, color=[255, 0, 0], type='', swap='kpt-7'),
6: dict(name='kpt-6', id=6, color=[255, 0, 0], type='', swap='kpt-4'),
7: dict(name='kpt-7', id=7, color=[255, 0, 0], type='', swap='kpt-5'),
8: dict(name='kpt-8', id=8, color=[255, 0, 0], type='', swap='kpt-9'),
9: dict(name='kpt-9', id=9, color=[255, 0, 0], type='', swap='kpt-8'),
10:
dict(name='kpt-10', id=10, color=[255, 0, 0], type='', swap='kpt-11'),
11:
dict(name='kpt-11', id=11, color=[255, 0, 0], type='', swap='kpt-10'),
12:
dict(name='kpt-12', id=12, color=[255, 0, 0], type='', swap='kpt-14'),
13:
dict(name='kpt-13', id=13, color=[255, 0, 0], type='', swap='kpt-15'),
14:
dict(name='kpt-14', id=14, color=[255, 0, 0], type='', swap='kpt-12'),
15:
dict(name='kpt-15', id=15, color=[255, 0, 0], type='', swap='kpt-13'),
16:
dict(name='kpt-16', id=16, color=[255, 0, 0], type='', swap='kpt-17'),
17:
dict(name='kpt-17', id=17, color=[255, 0, 0], type='', swap='kpt-16'),
18:
dict(name='kpt-18', id=18, color=[255, 0, 0], type='', swap='kpt-19'),
19:
dict(name='kpt-19', id=19, color=[255, 0, 0], type='', swap='kpt-18'),
20: dict(name='kpt-20', id=20, color=[255, 0, 0], type='', swap=''),
21: dict(name='kpt-21', id=21, color=[255, 0, 0], type='', swap=''),
22:
dict(name='kpt-22', id=22, color=[255, 0, 0], type='', swap='kpt-23'),
23:
dict(name='kpt-23', id=23, color=[255, 0, 0], type='', swap='kpt-22'),
24: dict(name='kpt-24', id=24, color=[255, 0, 0], type='', swap=''),
25: dict(name='kpt-25', id=25, color=[255, 0, 0], type='', swap=''),
26: dict(name='kpt-26', id=26, color=[255, 0, 0], type='', swap=''),
27: dict(name='kpt-27', id=27, color=[255, 0, 0], type='', swap=''),
28: dict(name='kpt-28', id=28, color=[255, 0, 0], type='', swap='')
},
skeleton_info={},
joint_weights=[1.] * 29,
sigmas=[])
configs/_base_/datasets/crowdpose.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='crowdpose',
paper_info=dict(
author='Li, Jiefeng and Wang, Can and Zhu, Hao and '
'Mao, Yihuan and Fang, Hao-Shu and Lu, Cewu',
title='CrowdPose: Efficient Crowded Scenes Pose Estimation '
'and A New Benchmark',
container='Proceedings of IEEE Conference on Computer '
'Vision and Pattern Recognition (CVPR)',
year='2019',
homepage='https://github.com/Jeff-sjtu/CrowdPose',
),
keypoint_info={
0:
dict(
name='left_shoulder',
id=0,
color=[51, 153, 255],
type='upper',
swap='right_shoulder'),
1:
dict(
name='right_shoulder',
id=1,
color=[51, 153, 255],
type='upper',
swap='left_shoulder'),
2:
dict(
name='left_elbow',
id=2,
color=[51, 153, 255],
type='upper',
swap='right_elbow'),
3:
dict(
name='right_elbow',
id=3,
color=[51, 153, 255],
type='upper',
swap='left_elbow'),
4:
dict(
name='left_wrist',
id=4,
color=[51, 153, 255],
type='upper',
swap='right_wrist'),
5:
dict(
name='right_wrist',
id=5,
color=[0, 255, 0],
type='upper',
swap='left_wrist'),
6:
dict(
name='left_hip',
id=6,
color=[255, 128, 0],
type='lower',
swap='right_hip'),
7:
dict(
name='right_hip',
id=7,
color=[0, 255, 0],
type='lower',
swap='left_hip'),
8:
dict(
name='left_knee',
id=8,
color=[255, 128, 0],
type='lower',
swap='right_knee'),
9:
dict(
name='right_knee',
id=9,
color=[0, 255, 0],
type='lower',
swap='left_knee'),
10:
dict(
name='left_ankle',
id=10,
color=[255, 128, 0],
type='lower',
swap='right_ankle'),
11:
dict(
name='right_ankle',
id=11,
color=[0, 255, 0],
type='lower',
swap='left_ankle'),
12:
dict(
name='top_head', id=12, color=[255, 128, 0], type='upper',
swap=''),
13:
dict(name='neck', id=13, color=[0, 255, 0], 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=('top_head', 'neck'), id=12, color=[51, 153, 255]),
13:
dict(link=('right_shoulder', 'neck'), id=13, color=[51, 153, 255]),
14:
dict(link=('left_shoulder', 'neck'), id=14, color=[51, 153, 255])
},
joint_weights=[
0.2, 0.2, 0.2, 1.3, 1.5, 0.2, 1.3, 1.5, 0.2, 0.2, 0.5, 0.2, 0.2, 0.5
],
sigmas=[
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.079, 0.079
])
configs/_base_/datasets/deepfashion2.py 0 → 100644
View file @ ca8a762a
This diff is collapsed.
configs/_base_/datasets/deepfashion_full.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='deepfashion_full',
paper_info=dict(
author='Liu, Ziwei and Luo, Ping and Qiu, Shi '
'and Wang, Xiaogang and Tang, Xiaoou',
title='DeepFashion: Powering Robust Clothes Recognition '
'and Retrieval with Rich Annotations',
container='Proceedings of IEEE Conference on Computer '
'Vision and Pattern Recognition (CVPR)',
year='2016',
homepage='http://mmlab.ie.cuhk.edu.hk/projects/'
'DeepFashion/LandmarkDetection.html',
),
keypoint_info={
0:
dict(
name='left collar',
id=0,
color=[255, 255, 255],
type='',
swap='right collar'),
1:
dict(
name='right collar',
id=1,
color=[255, 255, 255],
type='',
swap='left collar'),
2:
dict(
name='left sleeve',
id=2,
color=[255, 255, 255],
type='',
swap='right sleeve'),
3:
dict(
name='right sleeve',
id=3,
color=[255, 255, 255],
type='',
swap='left sleeve'),
4:
dict(
name='left waistline',
id=0,
color=[255, 255, 255],
type='',
swap='right waistline'),
5:
dict(
name='right waistline',
id=1,
color=[255, 255, 255],
type='',
swap='left waistline'),
6:
dict(
name='left hem',
id=2,
color=[255, 255, 255],
type='',
swap='right hem'),
7:
dict(
name='right hem',
id=3,
color=[255, 255, 255],
type='',
swap='left hem'),
},
skeleton_info={},
joint_weights=[1.] * 8,
sigmas=[])
configs/_base_/datasets/deepfashion_lower.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='deepfashion_lower',
paper_info=dict(
author='Liu, Ziwei and Luo, Ping and Qiu, Shi '
'and Wang, Xiaogang and Tang, Xiaoou',
title='DeepFashion: Powering Robust Clothes Recognition '
'and Retrieval with Rich Annotations',
container='Proceedings of IEEE Conference on Computer '
'Vision and Pattern Recognition (CVPR)',
year='2016',
homepage='http://mmlab.ie.cuhk.edu.hk/projects/'
'DeepFashion/LandmarkDetection.html',
),
keypoint_info={
0:
dict(
name='left waistline',
id=0,
color=[255, 255, 255],
type='',
swap='right waistline'),
1:
dict(
name='right waistline',
id=1,
color=[255, 255, 255],
type='',
swap='left waistline'),
2:
dict(
name='left hem',
id=2,
color=[255, 255, 255],
type='',
swap='right hem'),
3:
dict(
name='right hem',
id=3,
color=[255, 255, 255],
type='',
swap='left hem'),
},
skeleton_info={},
joint_weights=[1.] * 4,
sigmas=[])
configs/_base_/datasets/deepfashion_upper.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='deepfashion_upper',
paper_info=dict(
author='Liu, Ziwei and Luo, Ping and Qiu, Shi '
'and Wang, Xiaogang and Tang, Xiaoou',
title='DeepFashion: Powering Robust Clothes Recognition '
'and Retrieval with Rich Annotations',
container='Proceedings of IEEE Conference on Computer '
'Vision and Pattern Recognition (CVPR)',
year='2016',
homepage='http://mmlab.ie.cuhk.edu.hk/projects/'
'DeepFashion/LandmarkDetection.html',
),
keypoint_info={
0:
dict(
name='left collar',
id=0,
color=[255, 255, 255],
type='',
swap='right collar'),
1:
dict(
name='right collar',
id=1,
color=[255, 255, 255],
type='',
swap='left collar'),
2:
dict(
name='left sleeve',
id=2,
color=[255, 255, 255],
type='',
swap='right sleeve'),
3:
dict(
name='right sleeve',
id=3,
color=[255, 255, 255],
type='',
swap='left sleeve'),
4:
dict(
name='left hem',
id=4,
color=[255, 255, 255],
type='',
swap='right hem'),
5:
dict(
name='right hem',
id=5,
color=[255, 255, 255],
type='',
swap='left hem'),
},
skeleton_info={},
joint_weights=[1.] * 6,
sigmas=[])
configs/_base_/datasets/exlpose.py 0 → 100644
View file @ ca8a762a
dataset_info = dict(
dataset_name='exlpose',
paper_info=dict(
author='Sohyun Lee, Jaesung Rim, Boseung Jeong, Geonu Kim,'
'ByungJu Woo, Haechan Lee, Sunghyun Cho, Suha Kwak',
title='Human Pose Estimation in Extremely Low-Light Conditions',
container='arXiv',
year='2023',
homepage='https://arxiv.org/abs/2303.15410',
),
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='')
},
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=('left_elbow', 'left_wrist'), id=4, color=[0, 255, 0]),
5: dict(
link=('right_shoulder', 'right_elbow'), id=5, color=[255, 128, 0]),
6:
dict(link=('right_elbow', 'right_wrist'), id=6, color=[255, 128, 0]),
7: dict(link=('neck', 'right_hip'), id=7, color=[51, 153, 255]),
8: dict(link=('neck', 'left_hip'), id=8, color=[51, 153, 255]),
9: dict(link=('right_hip', 'right_knee'), id=9, color=[255, 128, 0]),
10:
dict(link=('right_knee', 'right_ankle'), id=10, color=[255, 128, 0]),
11: dict(link=('left_hip', 'left_knee'), id=11, color=[0, 255, 0]),
12: dict(link=('left_knee', 'left_ankle'), id=12, color=[0, 255, 0]),
},
joint_weights=[
0.2, 0.2, 0.2, 1.3, 1.5, 0.2, 1.3, 1.5, 0.2, 0.2, 0.5, 0.2, 0.2, 0.5
],
sigmas=[
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.079, 0.079
])
configs/_base_/datasets/fly.py 0 → 100644
View file @ ca8a762a
This diff is collapsed.
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