File size: 726 Bytes
5fee096 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | from .resnet import *
from .vit import vit_pt_imnet
from .vit import vit_pt_imnet_in21k_adapter
from .vit import vit_cl_lora
from .vit_dap import vit_pt_imnet_dap
from .SiNet import SiNet_vit
from .resnet_cbam import *
from .alexnet import AlexNet_TRGP, AlexNet_API
from .clip import clip
def get_backbone(config):
"""
Get the backbone according to the config dict.
Args:
config: The config dict.
Returns: The backbone module.
"""
kwargs = dict()
kwargs.update(config['backbone']['kwargs'])
try:
emb_func = eval(config["backbone"]['name'])(**kwargs)
except NameError:
raise ("{} is not implemented".format(config["backbone"]['name']))
return emb_func
|