| 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) |