cvt_labelmap_c17.py

Go to the documentation of this file.

import copy
 
labelmap_v10={
    0 : 'walk',
    1 : 'stand',
    2 : 'sit-Crouch',
    3 : 'sit-sofa',
    4 : 'sit-chair',
    5 : 'sit-floor',
    6 : 'falling-falling',
    7 : 'lie',
    8 : 'punch',
    9 : 'kick',
    10 : 'falling-fight',
    11 : 'plunging',
    12 : 'sit',
    13 : 'falling'
}
 
labelmap_v20={
    0:     "없음",
    1:     "앉기-바닥",
    2:     "앉기-의자",
    3:     "구부리기",
    4:     "걷기-혼자",
    5:     "걷기-잡고",
    6:     "걷기-앉은상태",
    7:     "걷기-계단오르기",
    8:     "걷기-계단내려가기",
    9:     "뛰기-달리기", 
    10:    "뛰기-외발뛰기",
    11:    "뛰기-갤로핑",
    12:    "뛰기-게걸음",
    13:    "일어서기-스스로",
    14:    "일어서기-짚고",
    15:    "눕기",
    16:    "눕기-업드려 머리들기",
    17:    "뒤집기-시도",
    18:    "뒤집기-성공",
    19:    "뒤집기-굴러가기",
    20:    "기어가기-무릎기기",
    21:    "기어가기-배밀이",
    22:    "기어가기-계단기기",
    23:    "기어오르기", 
    24:    "점프-제자리",
    25:    "점프-두발",
    26:    "잡기",
    27:    "끼적이기",
    28:    "빨기-손",
    29:    "빨기-놀잇감",
    30:    "안기-안겨있기",
    31:    "안기-안고있기",
    32:    "빼앗기",
    33:    "먹기-스스로",
    34:    "먹기-마시기",
    35:    "먹기-받아먹기",
    36:    "먹여주기",
    37:    "넘어짐", 
    38:    "부딪힘-벽", 
    39:    "부딪힘-사람",
    40:    "할퀴기",
    41:    "물기",
    42:    "떨어짐", 
    43:    "물건 던지기",
    44:    "손으로 때리기",
    45:    "물건으로 때리기",
    46:    "발로 때리기", 
    47:    "양팔제압하기", 
    48:    "밀치기", 
    49:    "당기기", 
    50:    "옷(기저귀) 갈아입기", 
    51:    "쓰다듬기", 
    52:    "토닥이기", 
    53:    "머리 따주기", 
    54:    "영유아를 들어서 옮기기", 
    55:    "물건 건네주기", 
    56:    "물건 받기", 
    57:    "사진찍기"
}
 
cvt_labelmap_v10_action_to_simple_ver=[
    [0 , 'walk', 6],
    [1 , 'stand', 4],
    [2 , 'sit-Crouch', 2],
    [3 , 'sit-sofa', 1],
    [4 , 'sit-chair', 1],
    [5 , 'sit-floor', 0],
    [6 , 'falling-falling', 14],
    [7 , 'lie', 9],
    [8 , 'punch', 15],
    [9 , 'kick', 16],
    [10 , 'falling-fight', 14],
]
 
#평가 레이블로 변환하는 테이블
#eval_table = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16]
eval_table  = [0, 0, 0, 0, 4, 4, 6, 7, 4, 9, 10, 10, 12, 13, 14, 15, 16]
 
#간소화 레이블
simple_lablemap={
    0 : "sit",#"앉기",
    1 : "sit_chair",
    2 : "sit_squat",
    3 : "sit_move", #앉은상태로 이동
    4 : "standing",#"서있기",
    5 : "bend_over",#허리숙이기
    6 : "walk",#"걷기",
    7 : "run",#"뛰기",
    #8 : "run_galloping", #갤로핑
    8 : "stand",#"일어서기",
    9 : "lying",#"누워있기",
    10 : "crawl",#"기어가기",
    11 : "crawl_pose",#"기어가기 자세",
    12 : "jump",#"점프",
    13 : "eating",#"먹기",
    14 : "fall-down",#"넘어짐", "떨어짐"    
    15 : "punch", #"펀치"
    16 : "kick", #"발차기",
    #18 : "scrawl", #"끼적이기"
}
 
cvt_labelmap_v21_action_to_simple_ver=[
    [0,     "없음", None],
    [1,     "앉기-바닥",None],
    [2,     "앉기-의자",None],
    [3,     "구부리기",None],
    [4,     "걷기-혼자",6],
    [5,     "걷기-잡고",6],
    [6,     "걷기-앉은상태",3],
    [7,     "걷기-계단오르기",6],
    [8,     "걷기-계단내려가기",6],
    [9,     "뛰기-달리기", 7],
    [10,    "뛰기-외발뛰기", 7],
    [11,    "뛰기-갤로핑", None],
    [12,    "뛰기-게걸음", None],
    [13,    "일어서기-스스로", 8],
    [14,    "일어서기-짚고", 8],
    [15,    "눕기", 9],
    [16,    "눕기-업드려 머리들기",9],
    [17,    "뒤집기-시도", 9],
    [18,    "뒤집기-성공", 9],
    [19,    "뒤집기-굴러가기", 9],
    [20,    "기어가기-무릎기기", 10],
    [21,    "기어가기-배밀이", None],
    [22,    "기어가기-계단기기", 10],
    [23,    "기어오르기", None],
    [24,    "점프-제자리", 12],
    [25,    "점프-두발", 12],
    [26,    "잡기", None],
    [27,    "끼적이기", None],
    [28,    "빨기-손", 13],
    [29,    "빨기-놀잇감", 13],
    [30,    "안기-안겨있기", None],
    [31,    "안기-안고있기", None],
    [32,    "빼앗기", None],
    [33,    "먹기-스스로", 13],
    [34,    "먹기-마시기", 13],
    [35,    "먹기-받아먹기", None],
    [36,    "먹여주기", None],
    [37,    "넘어짐", 14],
    [38,    "부딪힘-벽", None],
    [39,    "부딪힘-사람", None],
    [40,    "할퀴기", None],
    [41,    "물기", None],
    [42,    "떨어짐", None],
    [43,    "물건 던지기", None],
    [44,    "손으로 때리기", 15],
    [45,    "물건으로 때리기", 15],
    [46,    "발로 때리기", 16],
    [47,    "양팔제압하기", None],
    [48,    "밀치기", None],
    [49,    "당기기", None],
    [50,    "옷(기저귀) 갈아입기", None],
    [51,    "쓰다듬기", None],
    [52,    "토닥이기", None],
    [53,    "머리 따주기", None],
    [54,    "영유아를 들어서 옮기기", None],
    [55,    "물건 건네주기", None],
    [56,    "물건 받기", None],
    [57,    "사진찍기", None],
    [58,    "식별불가", -1]
]
 
cvt_labelmap_v21_foot_to_simple_ver=[
    [0,     "없음", None],
    [1,     "이동-느리게",6],
    [2,     "이동-빠르게",7],
    [3,     "이동-물건 위로",None],
    [4,     "이동-계단 오르기",None],
    [5,     "이동-계산 내려가기",None],
]
 
cvt_labelmap_v21_pose_to_simple_ver=[
    [0, "앉기-바닥", 0],
    [1, "앉기-의자", 1],
    [2, "앉기-무릎꿇기", 0],
    [3, "앉기-무릎서기", 4],
    [4, "앉기-쪼그리기", 2],
    [5, "서있기", 4],
    [6, "서있기-한발들기", 4],
    [7, "서있기-무릎구부리기", 4],
    [8, "서있기-허리구부리기", 5],
    [9, "누워있기-천장보고", 9],
    [10, "누워있기-엎드리기", 9],
    [11, "누워있기-옆으로", 9],
    [12, "누워있기-엎드려 머리들기", 9],
    [13, "누워있기-구부리기", 9],
    [14, "무릎기기 자세", 11]
]
 
simple_mutual_labelmap={
    0 : "None",
    1 : "안기",
    2 : "옷 갈아입히기(서서)",
    3 : "옷 갈아입히기(눕혀서)",
    4 : "쓰다듬기",
    5 : "토닥이기",
    6 : "머리따주기",
    7 : "먹여주기",
    8 : "물건 전달",
    9 : "밀기",
    10 : "당기기",
    11 : "부딪힘",
    12 : "손으로 때리기",
    13 : "발로 때리기",
    14 : "물기"
}
 
mutual_action_lablemap={
    0 : "None",
    1 : "안겨있기", #
    2 : "안고있기", #
    3 : "영유아를 들어서 이동", #
    4 : "옷(기저귀) 갈아입히기", #
    5 : "쓰다듬기", #
    6 : "토닥이기", #
    7 : "머리따주기",
    8 : "먹여주기", #
    9 : "물건 건네주기", #
    10 : "물건 받기", #
    11 : "장난감 빼앗기",
    12 : "손으로 때리기",
    13 : "물건으로 때리기",
    14 : "발로 때리기",
    15 : "신체 밀치기", #
    16 : "신체 당기기", #
    17 : "친구와 부딪힘",
    18 : "할퀴기",
    19 : "물기"
}
 
 
def cvt_labelmap_v21(annotations, cvt_action_info, cvt_pose_info):
    ret_annotations = copy.deepcopy(annotations)
    for i, anno in enumerate(annotations):
        action_id = anno['action_id']['action']
        new_action_id = cvt_action_info[action_id][2]
        
        if action_id == 11 or action_id == 12:
            move_id = anno['action_id']['foot']
            new_action_id = cvt_labelmap_v21_foot_to_simple_ver[move_id][2]
        elif new_action_id is None:
            pose_id = anno['action_id']['pose']
            new_action_id = cvt_pose_info[pose_id][2]
        
        if anno['truncation'] > 2 or anno['occlusion'] > 2:
            new_action_id = -1
            
        ret_annotations[i]['action_id']['action'] = new_action_id
        
    return ret_annotations
 
def cvt_labelmap_v10_pkl(pkl_data, cvt_action_info):
    ret_pkl = copy.deepcopy(pkl_data)
    for i, anno in enumerate(pkl_data):
        action_id = anno['label']
        new_action_id = cvt_action_info[action_id][2]
        
        ret_pkl[i]['label'] = new_action_id
        
    return ret_pkl