from .encoder_default import encoder_default def get_encoder(image_height, image_width, scale=0.25, sigma=1.5, encoder_type='default'): if encoder_type == 'default': encoder = encoder_default(image_height, image_width, scale, sigma) else: raise NotImplementedError return encoder