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