basicsr.utils.registry

class basicsr.utils.registry.Registry(name)

Bases: object

The registry that provides name -> object mapping, to support third-party users’ custom modules.

To create a registry (e.g. a backbone registry):

BACKBONE_REGISTRY = Registry('BACKBONE')

To register an object:

@BACKBONE_REGISTRY.register()
class MyBackbone():
    ...

Or:

BACKBONE_REGISTRY.register(MyBackbone)

get(name, suffix='basicsr')

keys()

register(obj=None, suffix=None)

Register the given object under the the name obj.__name__. Can be used as either a decorator or not. See docstring of this class for usage.