pyiqa.utils.registry

Module Contents

class pyiqa.utils.registry.Registry(name)

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)

register(obj=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.

get(name)

keys()

pyiqa.utils.registry.DATASET_REGISTRY

pyiqa.utils.registry.ARCH_REGISTRY

pyiqa.utils.registry.MODEL_REGISTRY

pyiqa.utils.registry.LOSS_REGISTRY

pyiqa.utils.registry.METRIC_REGISTRY