data/coco-kp.yaml

# 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]