Class ov::op::v13::NMSRotated#

class NMSRotated : public ov::op::Op#

NMSRotated operation.

Public Functions

NMSRotated(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 bool sort_result_descending = true, const ov::element::Type &output_type = ov::element::i64, const bool clockwise = true)#

Constructs a NMSRotated operation.

Parameters:

  • boxes – Node containing the coordinates of the bounding boxes

  • scores – Node containing the scores of the bounding boxes

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

  • iou_threshold – Node containing intersection over union threshold

  • score_threshold – Node containing minimum score threshold

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

  • output_type – Specifies the output type of the first and third output

  • clockwise – Specifies the direction of the rotation

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.