Class ov::op::v8::DetectionOutput

class DetectionOutput : public ov::op::util::DetectionOutputBase

Layer which performs non-max suppression to generate detection output using location and confidence predictions.

Public Functions

DetectionOutput(const Output<Node> &box_logits, const Output<Node> &class_preds, const Output<Node> &proposals, const Output<Node> &aux_class_preds, const Output<Node> &aux_box_preds, const Attributes &attrs)

Constructs a DetectionOutput operation.

Parameters

  • box_logits – Box logits

  • class_preds – Class predictions

  • proposals – Proposals

  • aux_class_preds – Auxilary class predictions

  • aux_box_preds – Auxilary box predictions

  • attrs – Detection Output attributes

DetectionOutput(const Output<Node> &box_logits, const Output<Node> &class_preds, const Output<Node> &proposals, const Attributes &attrs)

Constructs a DetectionOutput operation.

Parameters

  • box_logits – Box logits

  • class_preds – Class predictions

  • proposals – Proposals

  • attrs – Detection Output attributes

virtual void validate_and_infer_types() override

Verifies that attributes and inputs are consistent and computes output shapes and element types. Must be implemented by concrete child classes so that it can be run any number of times.

Throws if the node is invalid.