Step1_VesselSeg/nets/backbones/backbones.py

from timm.models import efficientnet


def build_backbone(model_name):
    model = getattr(Backbones, model_name)()
    return model


class Backbones(object):

    @staticmethod
    def efficientnet_b3_p():
        # 24, 12, 40, 120, 384
        model = efficientnet.efficientnet_b3_pruned(pretrained=False, features_only=True)
        return model