classification_utils.py

codes = ['', ':grinning_face:', ':smiling_face_with_smiling_eyes:', ':white_smiling_face:',
         ':face_savouring_delicious_food:', ':grinning_face_with_smiling_eyes:', ':neutral_face:',
         ':face_with_stuck-out_tongue:', ':face_with_stuck-out_tongue_and_winking_eye:',
         ':face_with_stuck-out_tongue_and_tightly-closed_eyes:', ':pensive_face:', ':face_throwing_a_kiss:',
         ':winking_face:', ':flushed_face:', ':loudly_crying_face:', ':angry_face:']


def lookup_label(code):
    return codes[code]


def lookup_code(label):
    return codes.index(label)


def group_faces(faces):
    if len(faces) < 2:
        return faces

    min_x = float('inf')
    min_y = float('inf')
    max_x = -1
    max_y = -1
    for face in faces:
        if face[0] < min_x:
            min_x = face[0]
        if face[1] < min_y:
            min_y = face[1]
        if face[0]+face[2] > max_x:
            max_x = face[0]+face[2]
        if face[1]+face[3] > max_y:
            max_y = face[1] + face[3]

    return [[min_x, min_y, max_x-min_x, max_y-min_y]]