CANN/ops-cv最近邻上采样算子

UpsampleNearest

【免费下载链接】ops-cv本项目是CANN提供的图像处理、目标检测相关的算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-cv

产品支持情况

功能说明

• 算子功能：

  对由多个输入通道组成的输入信号应用最近邻插值算法进行上采样。

  • 如果输入shape为(N, C, L)，则输出shape为(N, C, outputSize)；
  • 如果输入shape为(N, C, H, W)，则输出shape为(N, C, outputSize[0], outputSize[1])。

• 计算公式：

  • 当exact_mode=true时：

    $$ h_{src} = min(floor((h_{dst} + 0.5) * scalesH), H - 1) $$

    $$ w_{src} = min(floor((w_{dst} + 0.5) * scalesW), W - 1) $$

    $$ out(N, C, h_{dst}, w_{dst}) = self(N, C, h_{src}, w_{src}) $$

  • 当exact_mode=false时：

    $$ h_{src} = min(floor(h_{dst} * scalesH), H - 1) $$

    $$ w_{src} = min(floor(w_{dst} * scalesW), W - 1) $$

    $$ out(N, C, h_{dst}, w_{dst}) = self(N, C, h_{src}, w_{src}) $$

参数说明

• Atlas 推理系列产品 ：输入和输出的数据类型不支持BFLOAT16。
• Kirin X90 处理器系列产品 、 Kirin 9030 处理器系列产品 ：不支持BFLOAT16。

约束说明

无

调用说明

【免费下载链接】ops-cv本项目是CANN提供的图像处理、目标检测相关的算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-cv