File size: 317 Bytes
17cd746 |
1 2 3 4 5 6 7 8 9 |
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
|