luxonis_ml.nn_archive.config_building_blocks.base_models.head_metadata.HeadClassificationMetadata

class documentation

class HeadClassificationMetadata(HeadMetadata):

View In Hierarchy

Metadata for a classification head.

Instance Variable classes Names of classes recognized by the model.
Instance Variable is_softmax Whether the output already contains softmax probabilities.
Instance Variable n_classes Number of classes recognized by the model.

Inherited from HeadMetadata:

Class Variable model_config Undocumented
Instance Variable postprocessor_path Optional path to the postprocessor.
classes: list[str] =

Names of classes recognized by the model.

is_softmax: bool =

Whether the output already contains softmax probabilities.

n_classes: int =

Number of classes recognized by the model.