Note

You are reading the documentation for MMClassification 0.x, which will soon be deprecated at the end of 2022. We recommend you upgrade to MMClassification 1.0 to enjoy fruitful new features and better performance brought by OpenMMLab 2.0. Check the installation tutorial, migration tutorial and changelog for more details.

Source code for mmcls.models.builder

# 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)


[docs]def build_backbone(cfg):
    """Build backbone."""
    return BACKBONES.build(cfg)


[docs]def build_neck(cfg):
    """Build neck."""
    return NECKS.build(cfg)


[docs]def build_head(cfg):
    """Build head."""
    return HEADS.build(cfg)


[docs]def build_loss(cfg):
    """Build loss."""
    return LOSSES.build(cfg)


[docs]def build_classifier(cfg):
    return CLASSIFIERS.build(cfg)