Function extract_pose_params

data_generation/blender_utils.py:154–183

Extract pose parameters organized by bone name. Args: armature_data: List of frame dictionaries Returns: Dictionary mapping bone names to arrays of poses [num_frames, 7]

(armature_data)

def extract_pose_params(armature_data):
    """
    Extract pose parameters organized by bone name.

    Args:
        armature_data: List of frame dictionaries

    Returns:
        Dictionary mapping bone names to arrays of poses [num_frames, 7]
    """
    if len(armature_data) == 0:
        return {}

    # Get all bone names from first frame
    bone_names = list(armature_data[0].keys())
    num_frames = len(armature_data)

    # Organize by bone
    pose_params = {}
    for bone_name in bone_names:
        bone_poses = []
        for frame in armature_data:
            if bone_name in frame:
                bone_poses.append(frame[bone_name])
            else:
                # Use identity if bone missing in frame
                bone_poses.append([1, 0, 0, 0, 0, 0, 0])
        pose_params[bone_name] = np.array(bone_poses)

    return pose_params


def load_object_model(object_name, object_model_dir):