Kapao / data /coco-kp.yaml
AK391
add files
e6e7cb5
# train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/]
path: data/datasets/coco
labels: kp_labels
train: kp_labels/img_txt/train2017.txt
val: kp_labels/img_txt/val2017.txt
test: kp_labels/img_txt/test2017.txt
train_annotations: annotations/person_keypoints_train2017.json
val_annotations: annotations/person_keypoints_val2017.json
test_annotations: annotations/image_info_test-dev2017.json
pose_obj: True # write pose object labels
nc: 18 # number of classes (person class + 17 keypoint classes)
num_coords: 34 # number of keypoint coordinates (x, y)
# class names
names: [ 'person', 'nose',
'left_eye', 'right_eye',
'left_ear', 'right_ear',
'left_shoulder', 'right_shoulder',
'left_elbow', 'right_elbow',
'left_wrist', 'right_wrist',
'left_hip', 'right_hip',
'left_knee', 'right_knee',
'left_ankle', 'right_ankle' ]
kp_bbox: 0.05 # keypoint object size (normalized by longest img dim)
kp_flip: [0, 2, 1, 4, 3, 6, 5, 8, 7, 10, 9, 12, 11, 14, 13, 16, 15] # for left-right keypoint flipping
kp_left: [1, 3, 5, 7, 9, 11, 13, 15] # left keypoints
kp_names_short:
0: 'n'
1: 'ley'
2: 'rey'
3: 'lea'
4: 'rea'
5: 'ls'
6: 'rs'
7: 'lel'
8: 'rel'
9: 'lw'
10: 'rw'
11: 'lh'
12: 'rh'
13: 'lk'
14: 'rk'
15: 'la'
16: 'ra'
# segments for plotting
segments:
1: [5, 6]
2: [5, 11]
3: [11, 12]
4: [12, 6]
5: [5, 7]
6: [7, 9]
7: [6, 8]
8: [8, 10]
9: [11, 13]
10: [13, 15]
11: [12, 14]
12: [14, 16]