Class ov::op::v1::NonMaxSuppression#

class NonMaxSuppression : public ov::op::Op#

Elementwise addition operation.

Public Functions

NonMaxSuppression(const Output<Node> &boxes, const Output<Node> &scores, const Output<Node> &max_output_boxes_per_class, const Output<Node> &iou_threshold, const Output<Node> &score_threshold, const BoxEncodingType box_encoding = BoxEncodingType::CORNER, const bool sort_result_descending = true)#

Constructs a NonMaxSuppression operation.

Parameters:
  • boxesNode producing the box coordinates

  • scoresNode producing the box scores

  • max_output_boxes_per_classNode producing maximum number of boxes to be selected per class

  • iou_thresholdNode producing intersection over union threshold

  • score_thresholdNode producing minimum score threshold

  • box_encoding – Specifies the format of boxes data encoding

NonMaxSuppression(const Output<Node> &boxes, const Output<Node> &scores, const BoxEncodingType box_encoding = BoxEncodingType::CORNER, const bool sort_result_descending = true)#

Constructs a NonMaxSuppression operation with default values for the last 3 inputs.

Parameters:
  • boxesNode producing the box coordinates

  • scoresNode producing the box coordinates

  • box_encoding – Specifies the format of boxes data encoding

  • sort_result_descending – Specifies whether it is necessary to sort selected boxes across batches

  • output_type – Specifies the output tensor type

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.