| |
| import warnings |
|
|
| from mmseg.registry import MODELS |
|
|
| BACKBONES = MODELS |
| NECKS = MODELS |
| HEADS = MODELS |
| LOSSES = MODELS |
| SEGMENTORS = MODELS |
|
|
|
|
| def build_backbone(cfg): |
| """Build backbone.""" |
| warnings.warn('``build_backbone`` would be deprecated soon, please use ' |
| '``mmseg.registry.MODELS.build()`` ') |
| return BACKBONES.build(cfg) |
|
|
|
|
| def build_neck(cfg): |
| """Build neck.""" |
| warnings.warn('``build_neck`` would be deprecated soon, please use ' |
| '``mmseg.registry.MODELS.build()`` ') |
| return NECKS.build(cfg) |
|
|
|
|
| def build_head(cfg): |
| """Build head.""" |
| warnings.warn('``build_head`` would be deprecated soon, please use ' |
| '``mmseg.registry.MODELS.build()`` ') |
| return HEADS.build(cfg) |
|
|
|
|
| def build_loss(cfg): |
| """Build loss.""" |
| warnings.warn('``build_loss`` would be deprecated soon, please use ' |
| '``mmseg.registry.MODELS.build()`` ') |
| return LOSSES.build(cfg) |
|
|
|
|
| def build_segmentor(cfg, train_cfg=None, test_cfg=None): |
| """Build segmentor.""" |
| if train_cfg is not None or test_cfg is not None: |
| warnings.warn( |
| 'train_cfg and test_cfg is deprecated, ' |
| 'please specify them in model', UserWarning) |
| assert cfg.get('train_cfg') is None or train_cfg is None, \ |
| 'train_cfg specified in both outer field and model field ' |
| assert cfg.get('test_cfg') is None or test_cfg is None, \ |
| 'test_cfg specified in both outer field and model field ' |
| return SEGMENTORS.build( |
| cfg, default_args=dict(train_cfg=train_cfg, test_cfg=test_cfg)) |
|
|
