（beta）torch_npu.contrib.module.ROIAlign

产品支持情况

[object Object]undefined

功能说明

使用NPU API进行ROIAlign。

函数原型

参数说明

计算参数

• output_size ([object Object]): (height, width)，表示输出的高和宽。

• spatial_scale ([object Object]): 按此参数值缩放输入框。

• sampling_ratio ([object Object]): 为每个输出样本采集的输入样本数。0表示密集采样。

• aligned ([object Object]): 如果值为False，使用Detectron中的原实现方式。如果值为True，可更准确地对齐结果。

  [object Object]

计算输入

• input_tensor([object Object]): 输入张量，格式为NCHW。
• rois([object Object]): roi框，2D张量，第二个维度size为5，第一列表示roi框的索引，其余4列为roi框的坐标。

返回值说明

[object Object]

ROIAlign计算结果。

调用示例