RoiAlign

功能

在每个RoI（Region of Interest）区域进行池化处理。

输入

三个输入：

x：tensor，4D输入，数据类型：float16、float。

rois：shape=(num_rois, 4)，数据类型：float16、float。

batch_indices ：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