File size: 358 Bytes
beb7843 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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
|