aclnnRoiAlign
产品支持情况
| 产品 | 是否支持 |
|---|---|
| [object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object] | √ |
| [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object] | √ |
| [object Object]Atlas 200I/500 A2 推理产品[object Object] | × |
| [object Object]Atlas 推理系列产品 [object Object] | √ |
| [object Object]Atlas 训练系列产品[object Object] | √ |
功能说明
算子功能：RoiAlign是一种池化层，用于非均匀输入尺寸的特征图，并输出固定尺寸的特征图。
函数原型
每个算子分为undefined，必须先调用“aclnnRoiAlignGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnRoiAlign”接口执行计算。
aclnnStatus aclnnRoiAlignGetWorkspaceSize(const aclTensor* self, const aclTensor* rois, const aclTensor* batchIndices, const char* mode, int outputHeight, int outputWidth, int samplingRatio, float spatialScale, aclTensor* out, uint64_t* workspaceSize, aclOpExecutor** executor)aclnnStatus aclnnRoiAlign(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, const aclrtStream stream)
aclnnRoiAlignGetWorkspaceSize
参数说明：
self(aclTensor*, 计算输入)：Device侧的aclTensor，数据类型支持FLOAT、FLOAT16，必须与rois/out数据类型一致。支持undefined，undefined支持NCHW。维度为4维，shape为(N, C, H, W)。
rois(aclTensor*, 计算输入)：感兴趣区域。Device侧的aclTensor，数据类型支持FLOAT、FLOAT16，必须与self/out数据类型一致。支持undefined，undefined支持ND。维度为2维，shape为(numRois, 4)。
batchIndices(aclTensor*, 计算输入)：Device侧的aclTensor，数据类型支持INT32。支持undefined，undefined支持ND。表示每batch对应图像的索引。维度为1维，shape为(numRois,)。
out(aclTensor*, 计算输出)：Device侧的aclTensor，数据类型支持FLOAT、FLOAT16，必须与self/rois数据类型一致。支持undefined，undefined支持NCHW。维度为4维，shape为(numRois, C, outputHeight, outputWidth)。
mode(char*, 计算输入)：支持"avg"和"max"。池化模式。
outputHeight(int, 计算输入)：输出图像的高度。整型，建议传值1。
outputWidth(int, 计算输入)：输出图像的宽度。整型，建议传值1。
samplingRatio(int, 计算输入)：用于计算每个输出元素在H和W方向上的采样频率，整型，建议传值0。
spatialScale(float, 计算输入)：乘法空间尺度因子，将ROI坐标从其输入空间尺度转换为池化时使用的尺度，即输入特征图X相对于输入图像的空间尺度。浮点型，建议传值1.0。
workspaceSize(uint64_t*, 出参)：返回需要在Device侧申请的workspace大小。
executor(aclOpExecutor**, 出参)：返回op执行器，包含了算子计算流程。
返回码：
aclnnStatus：返回状态码，具体参见undefined。
aclnnRoiAlign
参数说明：
workspace(void*, 入参)：在Device侧申请的workspace内存地址。
workspaceSize(uint64_t, 入参)：在Device侧申请的workspace大小，由第一段接口aclnnRoiAlignGetWorkspaceSize获取。
executor(aclOpExecutor*, 入参)：op执行器，包含了算子计算流程。
stream(aclrtStream, 入参)：指定执行任务的Stream。
返回码：
aclnnStatus：返回状态码，具体参见undefined。
约束说明
无。
调用示例
示例代码如下，仅供参考，具体编译和执行过程请参考undefined。