metrics/registry.py

class Registry(object):
    def __init__(self):
        self.data = {}

    def register_module(self, module_name=None):
        def _register(cls):
            name = module_name
            if module_name is None:
                name = cls.__name__
            self.data[name] = cls
            return cls
        return _register

    def __getitem__(self, key):
        return self.data[key]

DETECTOR = Registry()
TRAINER  = Registry()
LOSSFUNC = Registry()