33 lines
771 B
Python
33 lines
771 B
Python
from custom_mmpkg.custom_mmcv.cnn import MODELS as MMCV_MODELS
|
|
from custom_mmpkg.custom_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)
|