15 lines
435 B
Python
15 lines
435 B
Python
from .transformer_encoder_fpn import *
|
|
try:
|
|
from .transformer_encoder_deform import *
|
|
except:
|
|
print('Deformable Transformer Encoder is not available.')
|
|
from .build import *
|
|
|
|
|
|
def build_encoder(config, *args, **kwargs):
|
|
model_name = config['MODEL']['ENCODER']['NAME']
|
|
|
|
if not is_model(model_name):
|
|
raise ValueError(f'Unkown model: {model_name}')
|
|
|
|
return model_entrypoints(model_name)(config, *args, **kwargs) |