Spaces:
Runtime error
Runtime error
| # Copyright (c) OpenMMLab. All rights reserved. | |
| 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)) | |