action_utils.py

Go to the documentation of this file.

import copy
 
 
def check_valid(pose_list, k):
    pose_num = len(pose_list)
    if pose_num < k*0.5:
        return False
    
    #이 조건은 좀더 고민 필요
    if pose_list[-1]['frame_id'] - pose_list[0]['frame_id'] < k-1:
        return False
 
    # k*0.5 프레임 이상 연속으로 미검출 됬을경우
    for i in range(1, pose_num):
        interval = pose_list[i]['frame_id'] - pose_list[i-1]['frame_id']
        if interval > k*0.5:
            return False
 
    return True
 
def make_dummy_pose(pose):
    #복사 필수
    tmp_pose = copy.deepcopy(pose)
    #스코어 0으로 만들기
    tmp_pose['keypoints'][:, 2] = 0
    return tmp_pose
 
def insert_dummy_pose(pose_list):
    pose_num = len(pose_list)
    ret_pose = [pose_list[0]]
    for i in range(1, pose_num):
        interval = pose_list[i]['frame_id'] - pose_list[i-1]['frame_id']
 
        #미검출 됬을경우 더미포즈 생성
        if interval > 1:
            ret_pose.extend([make_dummy_pose(pose_list[i])] * (interval-1))
        ret_pose.append(pose_list[i])
 
    return ret_pose