openvino.runtime.opset5.roi_align#

openvino.runtime.opset5.roi_align(data: Node | int | float | ndarray, rois: Node | int | float | ndarray, batch_indices: Node | int | float | ndarray, pooled_h: int, pooled_w: int, sampling_ratio: int, spatial_scale: float, mode: str, name: str | None = None) → Node#

Return a node which performs ROIAlign.

Parameters:

  • data – Input data.

  • rois – RoIs (Regions of Interest) to pool over.

  • batch_indices – Tensor with each element denoting the index of the corresponding image in the batch.

  • pooled_h – Height of the ROI output feature map.

  • pooled_w – Width of the ROI output feature map.

  • sampling_ratio – Number of bins over height and width to use to calculate each output feature map element.

  • spatial_scale – Multiplicative spatial scale factor to translate ROI coordinates.

  • mode – Method to perform pooling to produce output feature map elements.

Returns:

The new node which performs ROIAlign