Class ov::op::v5::NonMaxSuppression#

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

NonMaxSuppression operation.

Public Functions

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

Constructs a NonMaxSuppression operation with default values in the last 4 inputs.

Parameters:

  • boxes – Node producing the box coordinates

  • scores – Node producing the box scores

  • 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

NonMaxSuppression(const Output<Node> &boxes, const Output<Node> &scores, const Output<Node> &max_output_boxes_per_class, const BoxEncodingType box_encoding = BoxEncodingType::CORNER, const bool sort_result_descending = true, const ov::element::Type &output_type = ov::element::i64)#

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

Parameters:

  • boxes – Node producing the box coordinates

  • scores – Node producing the box scores

  • max_output_boxes_per_class – Node producing maximum number of boxes to be selected per class

  • 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

NonMaxSuppression(const Output<Node> &boxes, const Output<Node> &scores, const Output<Node> &max_output_boxes_per_class, const Output<Node> &iou_threshold, const BoxEncodingType box_encoding = BoxEncodingType::CORNER, const bool sort_result_descending = true, const ov::element::Type &output_type = ov::element::i64)#

Constructs a NonMaxSuppression operation with default values in the last. 2 inputs.

Parameters:

  • boxes – Node producing the box coordinates

  • scores – Node producing the box scores

  • max_output_boxes_per_class – Node producing maximum number of boxes to be selected per class

  • iou_threshold – Node producing intersection over union threshold

  • 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

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, const ov::element::Type &output_type = ov::element::i64)#

Constructs a NonMaxSuppression operation with default value in the last. input.

Parameters:

  • boxes – Node producing the box coordinates

  • scores – Node producing the box scores

  • max_output_boxes_per_class – Node producing maximum number of boxes to be selected per class

  • iou_threshold – Node producing intersection over union threshold

  • score_threshold – Node producing minimum score threshold

  • 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

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 Output<Node> &soft_nms_sigma, const BoxEncodingType box_encoding = BoxEncodingType::CORNER, const bool sort_result_descending = true, const ov::element::Type &output_type = ov::element::i64)#

Constructs a NonMaxSuppression operation.

Parameters:

  • boxes – Node producing the box coordinates

  • scores – Node producing the box scores

  • max_output_boxes_per_class – Node producing maximum number of boxes to be selected per class

  • iou_threshold – Node producing intersection over union threshold

  • score_threshold – Node producing minimum score threshold

  • soft_nms_sigma – Node specifying the sigma parameter for Soft-NMS

  • 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.