Function transform_joint_to_other_db_batch

util/transforms.py:103–120

(src_joint, src_name, dst_name)

def transform_joint_to_other_db_batch(src_joint, src_name, dst_name):
    
    src_joint_num = len(src_name)
    dst_joint_num = len(dst_name)
    person_num = src_joint.shape[0]
    new_joint = np.zeros(((
        person_num,
        dst_joint_num,
    ) + src_joint.shape[2:]),
                         dtype=np.float32)

    for src_idx in range(len(src_name)):
        name = src_name[src_idx]
        if name in dst_name:
            dst_idx = dst_name.index(name)
            new_joint[:, dst_idx] = src_joint[:, src_idx]

    return new_joint


def rot6d_to_axis_angle(x):