luxonis_ml.nn_archive.config_building_blocks.base_models.head_metadata.HeadObjectDetectionSSDMetadata

class documentation

class HeadObjectDetectionSSDMetadata(HeadObjectDetectionMetadata):

View In Hierarchy

Metadata for an SSD object detection head.

Instance Variable boxes_outputs Output containing predicted bounding box coordinates.
Instance Variable scores_outputs Output containing predicted bounding box confidence scores.

Inherited from HeadObjectDetectionMetadata:

Instance Variable anchors Optional predefined boxes of different sizes and aspect ratios. The innermost values describe box sizes, the middle values group anchors per output, and the outer values are ordered from smallest to largest output.
Instance Variable classes Names of classes detected by the model.
Instance Variable conf_threshold Confidence threshold above which a detection is considered valid.
Instance Variable iou_threshold Non-maximum suppression IoU threshold for filtering overlapping boxes.
Instance Variable max_det Maximum detections per image.
Instance Variable n_classes Number of classes detected by the model.

Inherited from HeadMetadata (via HeadObjectDetectionMetadata):

Class Variable model_config Undocumented
Instance Variable postprocessor_path Optional path to the postprocessor.
boxes_outputs: str =

Output containing predicted bounding box coordinates.

scores_outputs: str =

Output containing predicted bounding box confidence scores.