yuandong513
feat: init
17cd746
raw
history blame contribute delete
317 Bytes
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