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