module onnxrt.ops_cpu.op_non_max_suppression_#

Short summary#

module mlprodict.onnxrt.ops_cpu.op_non_max_suppression_

Implements runtime for operator NonMaxSuppression. The code is inspired from non_max_suppression.cc in onnxruntime.

source on GitHub

Classes#

class

truncated documentation

RuntimeNonMaxSuppression

Implements runtime for operator NonMaxSuppression. The code is inspired from non_max_suppression.cc …

Documentation#

Implements runtime for operator NonMaxSuppression. The code is inspired from non_max_suppression.cc in onnxruntime.

class mlprodict.onnxrt.ops_cpu.op_non_max_suppression_.RuntimeNonMaxSuppression(self: mlprodict.onnxrt.ops_cpu.op_non_max_suppression_.RuntimeNonMaxSuppression)#

Bases: pybind11_object

Implements runtime for operator NonMaxSuppression. The code is inspired from non_max_suppression.cc in onnxruntime.

__init__(self: mlprodict.onnxrt.ops_cpu.op_non_max_suppression_.RuntimeNonMaxSuppression) → None#
compute(self: mlprodict.onnxrt.ops_cpu.op_non_max_suppression_.RuntimeNonMaxSuppression, boxes: numpy.ndarray[numpy.float32], scores: numpy.ndarray[numpy.float32], max_output_boxes_per_class: numpy.ndarray[numpy.int64], iou_threshold: numpy.ndarray[numpy.float32], score_threshold: numpy.ndarray[numpy.float32]) → numpy.ndarray[numpy.int64]#

Computes NonMaxSuppression.

init(self: mlprodict.onnxrt.ops_cpu.op_non_max_suppression_.RuntimeNonMaxSuppression, center_point_box: int) → None#

initialization