openvino.runtime.opset14.nms_rotated

openvino.runtime.opset14.nms_rotated(boxes: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], scores: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], max_output_boxes_per_class: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], iou_threshold: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], score_threshold: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], sort_result_descending: bool = True, output_type: str = 'i64', clockwise: bool = True, name: Optional[str] = None) → openvino._pyopenvino.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