openvino.runtime.opset15.nms_rotated#

openvino.runtime.opset15.nms_rotated(boxes: Node | int | float | ndarray, scores: Node | int | float | ndarray, max_output_boxes_per_class: Node | int | float | ndarray, iou_threshold: Node | int | float | ndarray, score_threshold: Node | int | float | ndarray, sort_result_descending: bool = True, output_type: str = 'i64', clockwise: bool = True, name: str | None = None) → Node#

Return a node which performs NMSRotated.

Parameters:

  • boxes – Tensor with box coordinates of floating point type and shape [num_batches, num_boxes, 5], where the last dimension is defined as [x_ctr, y_ctr, width, height, angle_radians].

  • scores – Tensor with box scores of floating point type and shape [num_batches, num_classes, num_boxes].

  • max_output_boxes_per_class – Tensor (scalar or 1D) of integer type, specifying maximum number of boxes to be selected per class.

  • iou_threshold – Tensor (scalar or 1D) of floating point type, specifying intersection over union threshold

  • score_threshold – Tensor (scalar or 1D) of floating point type, specifying minimum score to consider box for the processing.

  • sort_result_descending – Flag that specifies whenever it is necessary to sort selected boxes across batches or not.

  • output_type – Output element type.

  • clockwise – Flag that specifies direction of the box rotation.

Returns:

The new node which performs NMSRotated