action_utils Namespace Reference

Functions
	check_valid (pose_list, k)
	make_dummy_pose (pose)
	insert_dummy_pose (pose_list)

Function Documentation

check_valid()

action_utils.check_valid	(		pose_list,
			k )

Definition at line 4 of file action_utils.py.

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
 

insert_dummy_pose()

action_utils.insert_dummy_pose

(

pose_list

)

Definition at line 28 of file action_utils.py.

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

make_dummy_pose()

action_utils.make_dummy_pose

(

pose

)

Definition at line 21 of file action_utils.py.

def make_dummy_pose(pose):
    #복사 필수
    tmp_pose = copy.deepcopy(pose)
    #스코어 0으로 만들기
    tmp_pose['keypoints'][:, 2] = 0
    return tmp_pose