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