RoiAlignRotated

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	×
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×
Kirin X90 处理器系列产品	√
Kirin 9030 处理器系列产品	√

功能说明

• 算子功能：用于旋转候选框的ROI对齐池化层。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入特征图。	FLOAT32	ND
rois	输入	ROI边界框。	FLOAT32	ND
spatial_scale	属性	ROI边界框的缩放率。	FLOAT32	-
sampling_ratio	属性	计算RoiAlign时的采样率。	INT	-
pooled_h	属性	RoiAlign输出时池化特征图的高。	INT	-
pooled_w	属性	RoiAlign输出时池化特征图的宽。	INT	-
aligned	属性	是否量化，true则将目标框中心坐标值减0.5。	BOOL	-
clockwise	属性	时钟方向，true为顺时针。	BOOL	-
output	输出	输出特征图。	FLOAT32	ND

约束说明

• 输入shape必须为[N,C,W,H]的Tensor，其中N、C、W和H均为INT32类型正整数。C的取值在(0, 1024]之间。
• rois的shape必须为[n, 6]的Tensor。n的取值范围为[1, 8192]。对于每个ROI而言，其组成为[batch_idx, center_x, center_y, w, h, angle]，其中batch_idx取值范围在[0, N)之间，会进行强制类型转换(float->int)，center_x、center_y、w、h为FLOAT32类型正浮点数，center_x与w的取值范围为[0, W)，center_y与h的取值范围为[0, H)，angle的取值为FLOAT32类型浮点数，取值范围为[0, π)。

调用说明

调用方式	样例代码	说明
图模式	test_geir_roi_align_rotated	通过算子IR接口方式调用RoiAlignRotated算子。