ngraph.opset4.non_max_suppression

ngraph.opset4.non_max_suppression(boxes: Union[_pyngraph.Node, int, float, numpy.ndarray], scores: Union[_pyngraph.Node, int, float, numpy.ndarray], max_output_boxes_per_class: Optional[Union[_pyngraph.Node, int, float, numpy.ndarray]] = None, iou_threshold: Optional[Union[_pyngraph.Node, int, float, numpy.ndarray]] = None, score_threshold: Optional[Union[_pyngraph.Node, int, float, numpy.ndarray]] = None, box_encoding: str = 'corner', sort_result_descending: bool = True, output_type: str = 'i64', name: Optional[str] = None)_pyngraph.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.

  • 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