[object Object]

[object Object][object Object]undefined[object Object]

• 接口功能：对输入特征图按 ROI（感兴趣区域）进行池化，在每个 ROI 内按空间划分为 [object Object] 个格子，对每个格子做最大池化，并输出池化结果及最大值在通道内的一维索引（argmax）。

• 计算公式：

  输入特征图 $x$ 的 shape 为 $(N, C, H, W)$，ROI 张量 $\text{rois}$ 的 shape 为 $(\text{num\_rois}, 5)$，每行表示 $(b_n, x_1, y_1, x_2, y_2)$。标量参数为 $s_h$、$s_w$（spatial_scale）以及 $\text{pooled\_h}$、$\text{pooled\_w}$。下标 $n$ 表示 ROI 索引，$c$ 表示通道，$(\text{ph}, \text{pw})$ 表示池化格点。

  • ROI 映射到特征图：将 ROI 坐标乘以 spatial_scale 得到特征图上的浮点区间：

    $$\tilde{x}_1 = x_1 s_w,\quad \tilde{y}_1 = y_1 s_h,\quad \tilde{x}_2 = (x_2+1)s_w,\quad \tilde{y}_2 = (y_2+1)s_h$$ $$W_{\text{roi}} = \tilde{x}_2 - \tilde{x}_1,\qquad H_{\text{roi}} = \tilde{y}_2 - \tilde{y}_1$$

    若 $W_{\text{roi}} \le 0$ 或 $H_{\text{roi}} \le 0$，该 ROI 的 $y$ 全为 0，$\text{argmax}$ 全为 -1。

  • Bin 步长与区间：每个池化格 (ph, pw) 对应 ROI 内一个 bin，步长与浮点区间为：

    $$\Delta w = \frac{W_{\text{roi}}}{\text{pooled\_w}},\qquad \Delta h = \frac{H_{\text{roi}}}{\text{pooled\_h}}$$ $$\tilde{w}_1 = \text{pw} \cdot \Delta w + \tilde{x}_1,\quad \tilde{w}_2 = (\text{pw}+1) \cdot \Delta w + \tilde{x}_1$$ $$\tilde{h}_1 = \text{ph} \cdot \Delta h + \tilde{y}_1,\quad \tilde{h}_2 = (\text{ph}+1) \cdot \Delta h + \tilde{y}_1$$

    取整并裁剪到 $[0,W) \times [0,H)$：

    $$w_1 = \text{clip}(\lfloor\tilde{w}_1\rfloor,\, 0,\, W),\quad w_2 = \text{clip}(\lceil\tilde{w}_2\rceil,\, 0,\, W)$$ $$h_1 = \text{clip}(\lfloor\tilde{h}_1\rfloor,\, 0,\, H),\quad h_2 = \text{clip}(\lceil\tilde{h}_2\rceil,\, 0,\, H)$$

    其中 $\text{clip}(a,l,u) = \min(\max(a,l), u)$。若 $w_2 \le w_1$ 或 $h_2 \le h_1$，该 bin 为空：$y=0$，$\text{argmax}=-1$。

  • 池化输出与 Argmax：记 $b = \text{rois}[n,0]$，bin 区域 $R = \{(h,w) : h_1 \le h < h_2,\, w_1 \le w < w_2\}$，则

    $$y[n,c,\text{ph},\text{pw}] = \max_{(h,w) \in R} x[b,c,h,w]$$

    （空 $R$ 时为 0。）

    $$\text{argmax}[n,c,\text{ph},\text{pw}] = h^* W + w^*$$

    $(h^*, w^*)$ 为 bin 内最大值位置（多解取第一个）；空 $R$ 为 -1。

  • 输出 Shape：

    [object Object]undefined

[object Object]

每个算子分为，必须先调用“aclnnRoiPoolingWithArgMaxGetWorkspaceSize”接口获取计算所需 workspace 大小以及包含了算子计算流程的执行器，再调用“aclnnRoiPoolingWithArgMax”接口执行计算。

[object Object]

• 参数说明：

  [object Object]

• 返回值：

  aclnnStatus：返回状态码，具体参见。

  第一段接口完成入参校验，出现以下场景时报错：

  [object Object]

[object Object]

• 参数说明：

  [object Object]

• 返回值：

  aclnnStatus：返回状态码，具体参见。

[object Object]

• 确定性计算：
  • aclnnRoiPoolingWithArgMax 默认确定性实现。

[object Object]

示例代码如下，仅供参考，具体编译和执行过程请参考。实际调用时需先通过 opgen 生成 [object Object]，若生成的头文件或接口签名不同，请以生成接口为准。