File size: 498 Bytes
caa56d6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
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]
BACKBONE = Registry()
DETECTOR = Registry()
TRAINER = Registry()
LOSSFUNC = Registry()
|