File size: 735 Bytes
373af33 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
from mmcv.cnn import MODELS as MMCV_MODELS
from mmcv.utils import Registry
def build_from_cfg(cfg, registry, default_args=None):
if cfg is None:
return None
return MMCV_MODELS.build_func(cfg, registry, default_args)
MODELS = Registry('models', parent=MMCV_MODELS, build_func=build_from_cfg)
LOSSES = MODELS
ARCHITECTURES = MODELS
SUBMODULES = MODELS
ATTENTIONS = MODELS
def build_loss(cfg):
"""Build loss."""
return LOSSES.build(cfg)
def build_architecture(cfg):
"""Build framework."""
return ARCHITECTURES.build(cfg)
def build_submodule(cfg):
"""Build submodule."""
return SUBMODULES.build(cfg)
def build_attention(cfg):
"""Build attention."""
return ATTENTIONS.build(cfg)
|