Function launch

pointersect/inference/main.py:2076–2138

launch the inference, using the parameters in the config yaml file. Args: config_filename: yaml filename

(
        config_filename: str,
)

def launch(
        config_filename: str,
):
    """
    launch the inference, using the parameters in the config yaml file.

    Args:
        config_filename:
            yaml filename
    """

    with open(config_filename) as file:
        config_dict = yaml.load(file, Loader=yaml.FullLoader)

    assert 'procedure' in config_dict
    procedure = config_dict['procedure']
    del config_dict['procedure']

    repo_root = os.path.normpath(os.path.join(__file__, '../../..'))


    # model filename
    default_model_pth_filename = os.path.normpath(
        os.path.join(
            repo_root, 'pointersect/checkpoint/epoch700.pth'))

    model_filename = config_dict['model_filename']
    if model_filename is None or model_filename == 'null':
        model_filename = default_model_pth_filename

    if isinstance(model_filename, str):
        model_filename = [model_filename]

    for i in range(len(model_filename)):
        if model_filename[i] is None or model_filename[i] == 'null':
            model_filename[i] = default_model_pth_filename
    config_dict['model_filename'] = model_filename

    if procedure == 'render_mesh':
        render_mesh(**config_dict)
    elif procedure == 'render_hypersim':
        render_hypersim(**config_dict)
    elif procedure == 'batch_render_mesh':
        batch_render_mesh(
            dataset_name=config_dict['dataset_name'],
            config_dict=config_dict,
        )
    elif procedure == 'render_arkitscenes':
        render_arkitscenes(
            **config_dict,
        )
    elif procedure == 'render_point_cloud':
        render_point_cloud(
            **config_dict,
        )
    elif procedure == 'render_point_clouds':
        render_point_clouds(
            **config_dict,