luxonis_ml.nn_archive.config_building_blocks.base_models.head_metadata.HeadSegmentationMetadata

class documentation

class HeadSegmentationMetadata(HeadMetadata):

View In Hierarchy

Metadata for a segmentation head.

Instance Variable background_class Whether the pixels classified as index 0 should be treated as background.
Instance Variable classes Names of classes segmented by the model.
Instance Variable is_softmax Whether the output already contains softmax probabilities.
Instance Variable n_classes Number of classes segmented by the model.

Inherited from HeadMetadata:

Class Variable model_config Undocumented
Instance Variable postprocessor_path Optional path to the postprocessor.
background_class: bool =

Whether the pixels classified as index 0 should be treated as background.

classes: list[str] =

Names of classes segmented by the model.

is_softmax: bool =

Whether the output already contains softmax probabilities.

n_classes: int =

Number of classes segmented by the model.