File size: 346 Bytes
5ceacbc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
_image_encoders = {}
def register_image_encoder(fn):
module_name_split = fn.__module__.split('.')
model_name = module_name_split[-1]
_image_encoders[model_name] = fn
return fn
def image_encoders(model_name):
return _image_encoders[model_name]
def is_image_encoder(model_name):
return model_name in _image_encoders
|