| from .focal import * | |
| from .focal_dw import * | |
| from .davit import * | |
| from .vit import * | |
| from .backbone import * | |
| from .build import * | |
| def build_backbone(config, **kwargs): | |
| model_name = config['MODEL']['BACKBONE']['NAME'] | |
| if not is_model(model_name): | |
| raise ValueError(f'Unkown model: {model_name}') | |
| return model_entrypoints(model_name)(config, **kwargs) |