예제 #1
0
def get_pose_extractor(name: str) -> PoseExtractor:
    r"""Fetches the correct pose_extractor using the given name and keyword arguments

    :param name: The name of the pose_extractor in the `habitat_sim.registry`
    """

    extractor = registry.get_pose_extractor(name)
    assert (extractor is not None
            ), "Could not find a pose extractor for name '{}'".format(name)

    return extractor
예제 #2
0
def make_pose_extractor(name: str) -> Callable[..., PoseExtractor]:
    r"""Constructs a pose_extractor using the given name and keyword arguments

    :param name: The name of the pose_extractor in the `habitat_sim.registry`
    :param kwargs: The keyword arguments to be passed to the constructor of the pose extractor
    """

    model = registry.get_pose_extractor(name)
    assert model is not None, "Could not find a pose extractor for name '{}'".format(
        name)

    return model