compressai_vision.registry

Register codecs, datasets, etc, to make them accessible via dynamic YAML configuration.

Registering maps a string name to a concrete creation method or class. This allows us to dynamically create an object depending on the given string name at runtime.

registry

compressai_vision.registry.registry.register_codec(name: str)

Decorator for registering a codec

compressai_vision.registry.registry.register_datacatalog(name: str)

Decorator for registering a dataset.

compressai_vision.registry.registry.register_dataset(name: str)

Decorator for registering a dataset.

compressai_vision.registry.registry.register_evaluator(name: str)

Decorator for registering an evaluator

compressai_vision.registry.registry.register_multask_codec(name: str)

Decorator for registering a multi-task codec

compressai_vision.registry.registry.register_pipeline(name: str)

Decorator for registering a pipeline

compressai_vision.registry.registry.register_vision_model(name: str)

Decorator for registering a vision model