luxonis_ml.nn_archive.config_building_blocks.base_models.Head

class documentation

class Head(BaseModel, ABC):

View In Hierarchy

Parser head definition for model outputs.

Instance Variable metadata Parser metadata.
Instance Variable name Optional name of the head.
Instance Variable outputs Optional names of model outputs fed into the parser. If omitted, all outputs are used.
Instance Variable parser Parser responsible for processing the model outputs.
metadata: HeadObjectDetectionMetadata | HeadClassificationMetadata | HeadObjectDetectionSSDMetadata | HeadSegmentationMetadata | HeadYOLOMetadata | HeadMetadata =

Parser metadata.

name: str | None =

Optional name of the head.

outputs: list[str] | None =

Optional names of model outputs fed into the parser. If omitted, all outputs are used.

parser: str =

Parser responsible for processing the model outputs.