support registry with basicsr suffix: can have repeated names in Basi…

…cSR and other repos

basicsr/utils/registry.py

@@ -35,12 +35,15 @@ def __init__(self, name):
  self._name = name
  self._obj_map = {}
 
- def _do_register(self, name, obj):
+ def _do_register(self, name, obj, suffix=None):
+ if isinstance(suffix, str):
+ name = name + '_' + suffix
+
  assert (name not in self._obj_map), (f"An object named '{name}' was already registered "
  f"in '{self._name}' registry!")
  self._obj_map[name] = obj
 
- def register(self, obj=None):
+ def register(self, obj=None, suffix=None):
  """
  Register the given object under the the name `obj.__name__`.
  Can be used as either a decorator or not.
@@ -50,17 +53,20 @@ def register(self, obj=None):
  # used as a decorator
  def deco(func_or_class):
  name = func_or_class.__name__
- self._do_register(name, func_or_class)
+ self._do_register(name, func_or_class, suffix)
  return func_or_class
 
  return deco
 
  # used as a function call
  name = obj.__name__
- self._do_register(name, obj)
+ self._do_register(name, obj, suffix)
 
- def get(self, name):
+ def get(self, name, suffix='basicsr'):
  ret = self._obj_map.get(name)
+ if ret is None:
+ ret = self._obj_map.get(name + '_' + suffix)
+ print(f'Name {name} is not found, use name: {name}_{suffix}!')
  if ret is None:
  raise KeyError(f"No object named '{name}' found in '{self._name}' registry!")
  return ret