openvino.runtime.opset14.non_max_suppression#
- openvino.runtime.opset14.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