文件最后提交记录最后更新时间
7 天前
17 天前
6 个月前
17 天前
17 天前
17 天前
6 个月前
1 个月前
README.md

UpsampleBilinear2d

产品支持情况

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

功能说明

  • 算子功能:对由多个输入通道组成的输入信号应用2D双线性上采样。
    • 对于输入shape:如果输入shape为(N, C, H, W),则输出shape为(N, C, outputSize[0], outputSize[1])。
    • 对于中心对齐的选择:一般像素被视为网格。当alignCorners = True时,像素被视为网格左上角的点,输出拐角处的像素与原图像的拐角像素中心对齐,同方向点之间是等间距的;当alignCorners = False时,像素被视为网格的交叉线上的点,输出拐角处的像素依然是原图像的拐角像素,但同方向点之间是不等距的。
  • 计算公式:
    • 核心算法逻辑:

      1. 将目标图像的每一个点映射回原图,得到一个带小数点的坐标。
      2. 根据这个浮点数坐标,计算前后相邻的原始图像的点。
      3. 分别计算相邻点到对应目标点的权重,按照权重相乘累加即可得到目标点值。
    • 具体计算逻辑: 缩放方式分为角对齐和边对齐,角对齐(alignCorners为true)表示按照原始图片左上角像素中心点对齐,边对齐(alignCorners为true)表示按照原始图片左上角顶点及两条边对齐,在计算缩放系数和坐标位置时存在差异。对于一个二维插值点(N,C,H,W)(N, C, H, W),则有以下公式:

      scale_h={(self.dim[2]−1)/(outputSize[0]−1)alignCorners=true1/scales_halignCorners=false&scales_h>0self.dim[2]/outputSize[0]alignCorners=falsescale\_h =\begin{cases} (self.dim[2]-1) / (outputSize[0]-1) & alignCorners=true \\ 1 / scales\_h & alignCorners=false\&scales\_h>0\\ self.dim[2] / outputSize[0] & alignCorners=false \end{cases}

      scale_w={(self.dim[3]−1)/(outputSize[1]−1)alignCorners=true1/scales_walignCorners=false&scales_w>0self.dim[3]/outputSize[1]alignCorners=falsescale\_w =\begin{cases} (self.dim[3]-1) / (outputSize[1]-1) & alignCorners=true \\ 1 / scales\_w & alignCorners=false\&scales\_w>0\\ self.dim[3] / outputSize[1] & alignCorners=false \end{cases}

      因此,对于output的某个方向上的点p(x,y),映射回原始图像中的点记为q(x',y'),则有关系:

      x′={x∗scale_halignCorners=trueMAX(0,(x+0.5)∗scale_h−0.5)alignCorners=falsex' =\begin{cases} x * scale\_h & alignCorners=true \\ MAX(0,{(x+0.5)*scale\_h-0.5}) & alignCorners=false \end{cases}

      y′={y∗scale_walignCorners=trueMAX(0,(y+0.5)∗scale_w−0.5)alignCorners=falsey' =\begin{cases} y * scale\_w & alignCorners=true \\ MAX(0,{(y+0.5)*scale\_w-0.5}) & alignCorners=false \end{cases}

      • 记:

        x0=int(x′),x1=int(x′)+1,lambda0=x1−x′,lambda1=1−lambda0x_{0} =int(x'),x_{1} =int(x')+1, lambda_{0} = x_{1}-x', lambda_{1} = 1-lambda_{0}

        y0=int(y′),y1=int(y′)+1,lambdb0=y1−y′,lambdb1=1−lambdb0y_{0} =int(y'),y_{1} =int(y')+1, lambdb_{0} = y_{1}-y', lambdb_{1} = 1-lambdb_{0}

      • 则有以下公式:

        V(px,y)=V(px0,y0)∗lambda0∗lambdb0+V(px0,y1)∗lambda0∗lambdb1+V(px1,y0)∗lambda1∗lambdb0+V(px1,y1)∗lambda1∗lambdb1{V(p_{x, y})} = {V(p_{x0, y0})} * {lambda_{0}} * {lambdb_{0}} + {V(p_{x0, y1})} * {lambda_{0}} * {lambdb_{1}} + {V(p_{x1, y0})} * {lambda_{1}} * {lambdb_{0}} + {V(p_{x1, y1})} * {lambda_{1}} * {lambdb_{1}}

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 表示进行上采样的输入张量,对应公式描述中的`self`。 FLOAT32、FLOAT16、BFLOAT16 ND
size 输入 指定输出空间大小,对应公式中的`outputSize`。size为2,表示指定`y`在H和W维度上的空间大小。 INT32 -
align_corners 可选属性
  • 决定是否对齐角像素点,对应公式中的`alignCorners`。align_corners为true,则输入和输出张量的角像素点会被对齐,否则输入和输出张量的左上角顶点及两条边对齐。
  • 默认值为false。
BOOL -
scales 可选属性
  • 指定空间大小的height和width维度乘数,对应公式中的`scales_h`、`scales_w`组成的数组[scales_h, scales_w]。
  • 默认值为空。
LISTFLOAT -
y 输出 表示采样后的输出张量,对应公式描述中的`output`。数据类型和数据格式与入参`x`的数据类型和数据格式保持一致。 FLOAT32、FLOAT16、BFLOAT16 ND
  • Kirin X90 处理器系列产品、Kirin 9030 处理器系列产品:不支持BFLOAT16。

约束说明

调用说明

调用方式 样例代码 说明
aclnn接口 test_aclnn_upsample_bilinear2d 通过aclnnUpsampleBilinear2d接口方式调用UpsampleBilinear2d算子。