功能
在每个RoI(Region of Interest)区域进行池化处理。
输入
- x:输入Tensor,4D输入,数据类型:float16、float。数据格式为NCHW,shape为(N, C, H, W)
- rois:感兴趣区域。shape=(num_rois, 4),数据类型:float16、float。
- batch_indices :batch对应图像的索引。shape=(num_rois,),数据类型:int64。
输出
y:输出tensor,和输入x同样的数据类型,shape=(num_rois, C, output_height, output_width)。
属性
- mode:string,默认为avg,含义:池化方式。
- output_height:int,默认为1,含义:y的高度。
- output_width:int,默认为1,含义:y的宽度。
- sampling_ratio :int,默认为0,含义:插值算法采样点数。
- spatial_scale:float,默认为1.0,含义:相对于输入图像的空间采样率。
- coordinate_transformation_mode : string,默认为half_pixel,含义:是否对输入值进行偏移(Opset v16及之后版本支持该属性)。
约束
不支持atc工具参数--precision_mode=must_keep_origin_dtype时float64的输入。
支持的ONNX版本
Opset v10/v11/v12/v13/v14/v15/v16/v17/v18