OpenOCR/opendet/losses/__init__.py

import copy
from importlib import import_module

name_to_module = {
    'DBLoss': '.db_loss',
}


def build_loss(config):
    config = copy.deepcopy(config)
    module_name = config.pop('name')
    assert module_name in name_to_module, Exception(
        '{} is not supported. The losses in {} are supportes'.format(
            module_name, list(name_to_module.keys())))

    module_path = name_to_module[module_name]
    module = import_module(module_path, package=__package__)
    module_class = getattr(module, module_name)

    return module_class(**config)