- from .resnet import build_resnet
- def build_backbone(cfg):
- print('==============================')
- print('Backbone: {}'.format(cfg['backbone']))
- # ResNet
- if "resnet" in cfg.backbone:
- return build_resnet(cfg)
- else:
- raise NotImplementedError("unknown backbone: {}".format(cfg.backbone))
-
-
|