openvino.runtime.opset6.roi_pooling

openvino.runtime.opset6.roi_pooling(input: Union[openvino.pyopenvino.Node, int, float, numpy.ndarray], coords: Union[openvino.pyopenvino.Node, int, float, numpy.ndarray], output_size: List[int], spatial_scale: Union[int, float, numpy.ndarray], method: str, name: Optional[str] = None) openvino.pyopenvino.Node

Return a node which produces an ROIPooling operation.

Parameters
  • input – Input feature map {N, C, …}.

  • coords – Coordinates of bounding boxes.

  • output_size – Height/Width of ROI output features (shape).

  • spatial_scale – Ratio of input feature map over input image size (float).

  • method – Method of pooling - string: “max” or “bilinear”.

Returns

ROIPooling node.