openvino.runtime.opset8.roi_align

openvino.runtime.opset8.roi_align(data: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], rois: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], batch_indices: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], pooled_h: int, pooled_w: int, sampling_ratio: int, spatial_scale: float, mode: str, name: Optional[str] = None) → openvino._pyopenvino.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