| from .registry import image_encoders | |
| from .registry import is_image_encoder | |
| def build_image_encoder(config_encoder, verbose, **kwargs): | |
| model_name = config_encoder['NAME'] | |
| if model_name.startswith('cls_'): | |
| model_name = model_name[4:] | |
| if not is_image_encoder(model_name): | |
| raise ValueError(f'Unkown model: {model_name}') | |
| return image_encoders(model_name)(config_encoder, verbose, **kwargs) | |