SMPLer-X/main/transformer_utils/mmpose/models/builder.py

# Copyright (c) OpenMMLab. All rights reserved.
from mmengine.registry import MODELS as MMCV_MODELS
from mmengine import Registry
from mmengine.registry import build_from_cfg, build_model_from_cfg

MODELS = Registry('models', parent=MMCV_MODELS, locations=['mmpose.models'])

BACKBONES = MODELS
NECKS = MODELS
HEADS = MODELS
LOSSES = MODELS
POSENETS = MODELS
MESH_MODELS = MODELS
TRANSFORMER = Registry('Transformer')


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_posenet(cfg):
    """Build posenet."""
    return POSENETS.build(cfg)


def build_mesh_model(cfg):
    """Build mesh model."""
    return MESH_MODELS.build(cfg)

def build_transformer(cfg, default_args=None):
    """Builder for Transformer."""
    return build_from_cfg(cfg, TRANSFORMER, default_args)