39 lines
750 B
Python
39 lines
750 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from mmcv.cnn import MODELS as MMCV_MODELS
|
|
from mmcv.cnn.bricks.registry import ATTENTION as MMCV_ATTENTION
|
|
from mmcv.utils import Registry
|
|
|
|
MODELS = Registry('models', parent=MMCV_MODELS)
|
|
|
|
BACKBONES = MODELS
|
|
NECKS = MODELS
|
|
HEADS = MODELS
|
|
LOSSES = MODELS
|
|
CLASSIFIERS = MODELS
|
|
|
|
ATTENTION = Registry('attention', parent=MMCV_ATTENTION)
|
|
|
|
|
|
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)
|