atomcloud.common.registry module

class atomcloud.common.registry.Registry(name)[source]

Bases: Iterable[Tuple[str, Any]]

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)
Parameters:

name (str) –

get(name)[source]
Parameters:

name (str) –

Return type:

Any

register(name, obj=None)[source]

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.

Parameters:

  • name (str) –

  • obj (Any | None) –

Return type:

Any