ModelOutput

Inheritance diagram of ModelOutput

ModelOutput inheritance diagram

class savant.base.model.ModelOutput(layer_names=<factory>, converter=None)

Model output parameters configuration template. Validates entries in a module config file under model.output.

model:
    output:
        layer_names: [output]
layer_names: List[str]

Specify model output layer names.

converter: PyFunc | None = None

Model output converter. Converter is used to transform raw tensor output into Savant data format.

Converter implementation should be written as a subclass of BaseObjectModelOutputConverter or BaseAttributeModelOutputConverter or BaseComplexModelOutputConverter depending on the model type.