昇腾社区首页
中文
注册
开发者
下载

aclnnUpsampleBilinear2dAA

产品支持情况

[object Object]undefined

功能说明

  • 接口功能:对由多个输入通道组成的输入信号应用2D双线性抗锯齿采样。

  • 计算公式:对于一个二维插值点(N,C,H,W)(N, C, H, W), 插值I(N,C,H,W)I(N, C, H, W)可以表示为:

    I(N,C,H,W)=i=0kWj=0kHw(i)w(j)f(hi,wj)/i=0kWw(i)/j=0kHw(j){I(N, C, H, W)} = \sum_{i=0}^{kW}\sum_{j=0}^{kH}{w(i) * w(j)} * {f(h_i, w_j)}/\sum_{i=0}^{kW}w(i)/\sum_{j=0}^{kH}w(j) scaleH={(input.dim(2)1)/(outputSize[0]1)alignCorners=true1/scalesHalignCorners=false&scalesH>0input.dim(2)/outputSize[0]otherwisescaleH =\begin{cases} (input.dim(2)-1) / (outputSize[0]-1) & alignCorners=true \\ 1 / scalesH & alignCorners=false\&scalesH>0\\ input.dim(2) / outputSize[0] & otherwise \end{cases} scaleW={(input.dim(3)1)/(outputSize[1]1)alignCorners=true1/scalesWalignCorners=false&scalesW>0input.dim(3)/outputSize[1]otherwisescaleW =\begin{cases} (input.dim(3)-1) / (outputSize[1]-1) & alignCorners=true \\ 1 / scalesW & alignCorners=false\&scalesW>0\\ input.dim(3) / outputSize[1] & otherwise \end{cases}

    其中:

    • kWkWkHkH分别表示W方向和H方向影响插值点大小的点的数量
    • 如果scaleH>=1scaleH >= 1,则kH=floor(scaleH)2+1kH = floor(scaleH) * 2 + 1,否则kH=3kH = 3
    • 如果scaleW>=1scaleW >= 1,则kW=floor(scaleW)2+1kW = floor(scaleW) * 2 + 1,否则kW=3kW = 3
    • f(hi,wj)f(h_i, w_j)是原图像在(hi,wj)(h_i, w_j)的像素值
    • w(i)w(i)w(j)w(j)是双线性抗锯齿插值的W方向和H方向权重,计算公式为:w(i)={1hihhih<10otherwisew(i) = \begin{cases} 1 - |h_i - h| & |h_i -h| < 1 \\ 0 & otherwise \end{cases} w(j)={1wjwwjw<10otherwisew(j) = \begin{cases} 1 - |w_j - w| & |w_j -w| < 1 \\ 0 & otherwise \end{cases}

函数原型

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

[object Object]
[object Object]

aclnnUpsampleBilinear2dAAGetWorkspaceSize

  • 参数说明

    [object Object]
  • 返回值

    aclnnStatus:返回状态码,具体参见

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

    [object Object]

aclnnUpsampleBilinear2dAA

  • 参数说明

    [object Object]
  • 返回值

aclnnStatus:返回状态码,具体参见

约束说明

  • 输入数据缩放场景缩小倍数必须小于等于50,即:

    输入shape的高度H/outputSize_H<=50输入shape的高度H/outputSize\_H <= 50 输入shape的宽度W/outputSize_W<=50输入shape的宽度W/outputSize\_W <=50
  • 参数input、outputSize、scalesH、scalesW需要满足如下约束:

    outputSize_H=floor(input_HscalesH)outputSize\_H = floor(input\_H * scalesH) outputSize_W=floor(input_WscalesW)outputSize\_W = floor(input\_W * scalesW)
  • 确定性计算:

    • aclnnUpsampleBilinear2dAA默认确定性实现。

调用示例

示例代码如下,仅供参考,具体编译和执行过程请参考

[object Object]