ngraph.roi_pooling

ngraph.roi_pooling(input: Union[_pyngraph.Node, int, float, numpy.ndarray], coords: Union[_pyngraph.Node, int, float, numpy.ndarray], output_size: List[int], spatial_scale: Union[int, float, numpy.ndarray], method: str, name: Optional[str] = None) _pyngraph.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