from .backbone_2d.backbone_2d import Backbone2D
from .backbone_3d.backbone_3d import Backbone3D


def build_backbone_2d(cfg, pretrained=False):
    backbone = Backbone2D(cfg, pretrained)
    return backbone, backbone.feat_dims


def build_backbone_3d(cfg, pretrained=False):
    backbone = Backbone3D(cfg, pretrained)
    return backbone, backbone.feat_dim