Shortcuts

备注

您正在阅读 MMClassification 0.x 版本的文档。MMClassification 0.x 会在 2022 年末被切换为次要分支。建议您升级到 MMClassification 1.0 版本,体验更多新特性和新功能。请查阅 MMClassification 1.0 的安装教程迁移教程以及更新日志

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)


[文档]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)
Read the Docs v: latest
Versions
master
latest
1.x
dev-1.x
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.