module `onnxrt.ops_cpu.op_roi_align_`#

Short summary#

module mlprodict.onnxrt.ops_cpu.op_roi_align_

Implements runtime for operator RoiAlign. The code is inspired from pool.cc in onnxruntime.

class	truncated documentation
`RoiAlignDouble`	Implements float runtime for operator RoiAlign. The code is inspired from pool.cc …
`RoiAlignFloat`	Implements float runtime for operator RoiAlign. The code is inspired from pool.cc …

Implements runtime for operator RoiAlign. The code is inspired from pool.cc in onnxruntime.

class mlprodict.onnxrt.ops_cpu.op_roi_align_.RoiAlignDouble(self: mlprodict.onnxrt.ops_cpu.op_roi_align_.RoiAlignDouble)#

Bases: pybind11_object

Implements float runtime for operator RoiAlign. The code is inspired from pool.cc in onnxruntime. Supports double only.

__init__(self: mlprodict.onnxrt.ops_cpu.op_roi_align_.RoiAlignDouble) → None#

compute(self: mlprodict.onnxrt.ops_cpu.op_roi_align_.RoiAlignDouble, arg0: numpy.ndarray[numpy.float64], arg1: numpy.ndarray[numpy.float64], arg2: numpy.ndarray[numpy.int64]) → numpy.ndarray[numpy.float64]#: Computes the output for operator RoiAlign.

init(self: mlprodict.onnxrt.ops_cpu.op_roi_align_.RoiAlignDouble, arg0: str, arg1: str, arg2: int, arg3: int, arg4: int, arg5: float) → None#: Initializes the runtime with the ONNX attributes.

class mlprodict.onnxrt.ops_cpu.op_roi_align_.RoiAlignFloat(self: mlprodict.onnxrt.ops_cpu.op_roi_align_.RoiAlignFloat)#

Bases: pybind11_object

Implements float runtime for operator RoiAlign. The code is inspired from pool.cc in onnxruntime. Supports float only.

__init__(self: mlprodict.onnxrt.ops_cpu.op_roi_align_.RoiAlignFloat) → None#

compute(self: mlprodict.onnxrt.ops_cpu.op_roi_align_.RoiAlignFloat, arg0: numpy.ndarray[numpy.float32], arg1: numpy.ndarray[numpy.float32], arg2: numpy.ndarray[numpy.int64]) → numpy.ndarray[numpy.float32]#: Computes the output for operator RoiAlign.

init(self: mlprodict.onnxrt.ops_cpu.op_roi_align_.RoiAlignFloat, arg0: str, arg1: str, arg2: int, arg3: int, arg4: int, arg5: float) → None#: Initializes the runtime with the ONNX attributes.