Function build_semantic_sam_head

semantic_sam/body/build.py:6–12

(config, *args, **kwargs)

def build_semantic_sam_head(config, *args, **kwargs):
    model_name = config['MODEL']['HEAD']
    if not is_model(model_name):
        raise ValueError(f'Unkown model: {model_name}')

    body = model_entrypoints(model_name)(config, *args, **kwargs)
    return body