35 lines
579 B
Python
35 lines
579 B
Python
from mmcv.cnn import MODELS as MMCV_MODELS
|
|
from mmcv.utils import Registry
|
|
|
|
MODELS = Registry('models', parent=MMCV_MODELS)
|
|
|
|
BACKBONES = MODELS
|
|
NECKS = MODELS
|
|
HEADS = MODELS
|
|
LOSSES = MODELS
|
|
CLASSIFIERS = MODELS
|
|
|
|
|
|
def build_backbone(cfg):
|
|
"""Build backbone."""
|
|
return BACKBONES.build(cfg)
|
|
|
|
|
|
def build_neck(cfg):
|
|
"""Build neck."""
|
|
return NECKS.build(cfg)
|
|
|
|
|
|
def build_head(cfg):
|
|
"""Build head."""
|
|
return HEADS.build(cfg)
|
|
|
|
|
|
def build_loss(cfg):
|
|
"""Build loss."""
|
|
return LOSSES.build(cfg)
|
|
|
|
|
|
def build_classifier(cfg):
|
|
return CLASSIFIERS.build(cfg)
|