RoiAlign

功能

在每个roi区域进行池化处理

边界

【输入】

3个输入

x:tensor,4D输入,数据类型:float16、float32

rois:shape=(num_rois, 4),数据类型:float16、float32

batch_indices :shape=(num_rois,),数据类型:int64

【输出】

一个输出

y:一个张量,和输入x同样的type,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,含义:相对于输入图像的空间采样率

【约束】

batch_indices数据类型只能写int32不能写int64

不支持atc工具参数--precision_mode=must_keep_origin_dtype时float32,float64的输入

支持的ONNX版本

Opset v10/v11/v12/v13