settings.py

import cv2
import numpy as np

DATA_PATH = 'data/keypoints/'
VIDEO_PATH = 'data/videos/'
POSE_PATH = 'data/search_query'

DATASET_VIDEOS = [
    'TB_F_FB',
    'UP3',
    'UP2',
    'UP1',
    'TR_S',
    'TR_F',
    'TOS_S',
    'TOS_F',
    'TL_S',
    'TL_F',
    'TF_S',
    'TF_F',
    'TB_S',
    'TB_S_FB',
    'TA',
    'SYN_U',
    'SYN_S',
    'SYN_R',
    'SYN_K',
    'SYN_B',
    'SS_S_RT',
    'SS_S_LT',
    'SS_S_FT',
    'SS_S_BK',
    'SS_F_RT',
    'SS_F_LT',
    'SS_F_FT',
    'SS_F_BK',
    'SJ_RT',
    'SJ_LT',
    'SJ_FT',
    'SJ_BK',
    'SB_R_WA',
    'SB_R_NA',
    'RV_R_S',
    'RV_R_F',
    'PV_R_S',
    'PV_R_F',
    'PE',
    'PC_F',
    'PC_S',
    'PB_S',
    'PB_F',
    'P3',
    'P2',
    'P1',
    'LU_S_dis',
    'LU_S_big',
    'LU_F_dis',
    'LU_F_big',
    'LD_S_small',
    'LD_S_dis',
    'LD_F_small',
    'FR_R_WA',
    'LD_F_dis',
    'FR_R_NA',
    'EF',
    'CU_R_WA',
    'CU_R_NA',
    'CH',
    'BSS_S_FT',
    'BSS_S_BK',
    'BS_S_RT',
    'BSS_F_FT',
    'BSS_F_BK',
    'BS_S_LT',
    'BS_S_FT',
    'BS_S_BK',
    'BS_F_RT',
    'BS_F_LT',
    'BS_F_FT',
    'BS_F_BK',
    'BJ_RT',
    'BJ_LT',
    'BJ_FT',
    'BJ_BK',
    'BBS_S_FT',
    'BBS_S_BK',
    'BBS_F_FT',
    'BBS_F_BK',
    'BA_R_WA',
    'BA_R_NA',
    'AS_L_WA',
    'AS_L_NA',
    'TB_F_FB',
    'AR',
]

COLORS = [
    'rgb(255, 0, 0)',
    'rgb(255, 0, 0)',
    'rgb(255, 170, 0)',
    'rgb(255, 255, 0)',
    'rgb(255, 85, 0)',
    'rgb(170, 255, 0)',
    'rgb(85, 255, 0)',
    'rgb(0, 255, 0)',
    'rgb(0, 255, 170)',
    'rgb(0, 255, 85)',
    'rgb(0, 170, 255)',
    'rgb(85, 0, 255)',
    'rgb(0, 85, 255)',
    'rgb(255, 0, 85)',
    'rgb(0, 0, 255)',
    'rgb(255, 0, 170)',
    'rgb(170, 0, 255)',
    'rgb(255, 0, 255)',
    'rgb(0, 0, 255)',
    'rgb(0, 0, 255)',
    'rgb(0, 0, 255)',
    'rgb(0, 255, 255)',
    'rgb(0, 255, 255)',
    'rgb(0, 255, 255)',
]


PAIRS_RENDER = np.array(
    [[1, 8], [1, 2], [1, 5], [2, 3], [3, 4], [5, 6], [6, 7], [8, 9], [9, 10], [10, 11], [8, 12], [12, 13], [13, 14],
     [1, 0], [0, 15], [15, 17], [0, 16], [16, 18], [14, 19], [19, 20], [14, 21], [11, 22], [22, 23], [11, 24]]
)


BODYPARTS_POINT = [
     "Nose",
     "Neck",
     "Right Shoulder",
     "Right Elbow",
     "Right Wrist",
     "Left Shoulder",
     "Left Elbow",
     "Left Wrist",
     "MidHip",
     "Right Hip",
     "Right Knee",
     "Right Ankle",
     "Left Hip",
     "Left Knee",
     "Left Ankle",
     "Right Eye",
     "Left Eye",
     "Right Ear",
     "Left Ear",
     "Left BigToe",
     "Left SmallToe",
     "Left Heel",
     "Right BigToe",
     "Right SmallToe",
      "Right Heel",
]


BODY_SEGMENTS = {
    'torso': [1,8],
    'right_shoulder': [1,2],
    'right_upper_arm': [2,3],
    'right_lower_arm': [3,4],
    'left_shoulder': [1,5],
    'left_upper_arm': [5,6],
    'left_lower_arm': [6,7],
    'right_hip': [8,9],
    'right_upper_leg': [9,10],
    'right_lower_leg': [10,11],
    'left_hip': [8,12],
    'left_upper_leg': [12,13],
    'left_lower_leg': [13,14],
    'nose_to_neck': [1,0],
    'right_eye_to_nose': [0,15],
    'right_ear_to_eye': [15,17],
    'left_eye_to_nose': [0,16],
    'left_ear_to_eye': [16,18],
    'left_foot': [14,19],
    'left_toes': [19,20],
    'left_ankle_to_heel': [14,21],
    'right_foot': [11,22],
    'right_toes': [22,23],
    'right_ankle_to_heel': [11,24],
}


BODYPART_INDEX = {
    0: 'nose_to_neck_to_left_shoulder',
    1: 'nose_to_neck_to_right_shoulder',
    2: 'left_shoulder_to_right_shoulder',
    3: 'left_shoulder_to_left_upper_arm',
    4: 'left_lower_arm_to_left_upper_arm',
    5: 'right_upper_arm_to_right_shoulder',
    6: 'right_upper_arm_to_right_lower_arm',
    7: 'left_eye_to_nose_to_left_ear_to_eye',
    8: 'left_eye_to_nose_to_neck',
    9: 'nose_to_neck_to_right_eye_to_nose',
    10: 'left_eye_to_nose_to_right_eye_to_nose',
    11: 'right_eye_to_nose_to_right_ear_to_eye',
    12: 'right_hip_to_right_upper_leg',
    13: 'right_upper_leg_to_right_lower_leg',
    14: 'left_hip_to_left_upper_leg',
    15: 'left_upper_leg_to_left_lower_leg',
    16: 'left_lower_leg_left_ankle_to_heel',
    17: 'right_lower_leg_to_right_ankle_to_heel',
    18: 'right_foot_to_right_toes',
    19: 'right_foot_to_right_lower_leg',
    20: 'right_foot_to_right_ankle_to_heel',
    21: 'left_foot_to_left_lower_leg',
    22: 'left_foot_to_left_ankle_to_heel',
    23: 'left_foot_to_left_toes',
    24: 'torso_to_right_shoulder',
    25: 'torso_to_left_shoulder',
    26: 'torso_to_nose_to_neck',
    27: 'torso_to_right_hip',
    28: 'torso_to_left_hip'
}


POSES_DICT = {
    'qsearch-1': {'src': 'LU_S_big_frame_52.png', 'data':'LU_S_big_000000000052_keypoints.json'},
    'qsearch-2': {'src': 'TOS_F_frame_38.png', 'data': 'TOS_F_000000000038_keypoints.json'},
    'qsearch-3': {'src': 'CU_R_NA_frame_6.png', 'data': 'CU_R_NA_000000000006_keypoints.json'},
}


LABEL_DICT = {
    'BS_F_BK': 1, 'BS_F_LT': 1, 'BS_F_RT': 1, 'BS_S_BK': 1, 'BS_S_FT': 1, 'BS_S_LT': 1,
    'BS_S_RT': 1, 'SS_F_BK': 1, 'SS_F_FT': 1, 'SS_F_LT': 1, 'SS_F_RT': 1, 'SS_S_BK': 1,
    'SS_S_LT': 1, 'SS_S_RT': 1, 'SYN_R': 1, 'SYN_S': 1,
    'BJ_BK': 2, 'BJ_FT': 2, 'LD_F_dis': 2, 'LD_F_small': 2, 'LD_S_dis': 2, 'LD_S_small': 2,
    'LU_F_big': 2, 'LU_F_dis': 2, 'LU_S_big': 2, 'LU_S_dis': 2, 'SJ_BK': 2, 'SJ_FT': 2,
    'BJ_RT': 3, 'SJ_RT': 3,
    'AR': 4, 'TA': 4,
    'BJ_LT': 5, 'SJ_LT': 5,
    'BBS_F_BK': 6, 'BBS_F_FT': 6, 'BBS_S_BK': 6, 'BBS_S_FT': 6, 'BSS_S_BK': 6,
    'BSS_S_FT': 6,
    'TB_S': 7, 'TB_S_FB': 7, 'TF_F': 7, 'TF_S': 7, 'TL_F': 7, 'TL_S': 7, 'TOS_F': 7, 'TOS_S': 7, 'TR_F': 7,
    'TR_S': 7,
    'SYN_K': 8,
    'TB_F_FB': 9
}