openvino.runtime.opset15.non_max_suppression#

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

Return a node which performs NonMaxSuppression.

Parameters:

  • boxes – Tensor with box coordinates.

  • scores – Tensor with box scores.

  • max_output_boxes_per_class – Tensor Specifying maximum number of boxes to be selected per class.

  • iou_threshold – Tensor specifying intersection over union threshold

  • score_threshold – Tensor specifying minimum score to consider box for the processing.

  • soft_nms_sigma – Tensor specifying the sigma parameter for Soft-NMS.

  • box_encoding – Format of boxes data encoding.

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

  • output_type – Output element type.

Returns:

The new node which performs NonMaxSuppression