File size: 371 Bytes
814a594 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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) |