DeformableConv2d

产品支持情况

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

功能说明

• 算子功能：实现卷积功能，支持2D卷积，同时支持可变形卷积、分组卷积。

• 计算公式：

  假定输入（input）的shape是[N, inH, inW, inC]，输出的（out）的shape为[N, outH, outC, outW]，根据已有参数计算outH、outW：

  $$outH=(inH+padding[0]+padding[1]-((K_H-1)\ast dilation[2]+1))//stride[2]+1$$

  $$outW=(inW+padding[2]+padding[3]-((K_W-1)\ast dilation[3]+1))//stride[3]+1$$

  标准卷积计算采样点下标：

  $$x=-padding[2]+ow\ast stride[3]+kw\ast dilation[3],kw的取值为(0,K\_W-1)$$

  $$y=-padding[0]+oh\ast stride[2]+kh\ast dilation[2],kh的取值为(0,K\_H-1)$$

  根据传入的offset，进行变形卷积，计算偏移后的下标：

  $$(x,y)=(x+offsetX,y+offsetY)$$

  使用双线性插值计算偏移后点的值：

  $$\begin{gathered} (x_0,y_0)=(int(x),int(y)) \\ (x_1,y_1)=(x_0+1,y_0+1) \end{gathered}$$

  $$\begin{gathered} weigh{t}_{00}=(x_1-x)\ast (y_1-y) \\ weigh{t}_{01}=(x_1-x)\ast (y-y_0) \\ weigh{t}_{10}=(x-x_0)\ast (y_1-y) \\ weigh{t}_{11}=(x-x_0)\ast (y-y_0) \end{gathered}$$

  $$deformOut(x,y)=weigh{t}_{00}\ast input(x0,y0)+weigh{t}_{01}\ast input(x0,y1)+weigh{t}_{10}\ast input(x1,y0)+weigh{t}_{11}\ast input(x1,y1)$$

  进行卷积计算得到最终输出：

  $$\text{out}(N_i,C_{{\text{out}}_j})=\sum _{k=0}^{C_{\text{in}}-1}\text{weight}(C_{{\text{out}}_j},k)\star \text{deformOut}(N_i,k)$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	表示输入的原始数据。对应公式中的`input`。shape为[N, inH, inW, inC]，其中inH * inW不能超过2147483647。	FLOAT32、FLOAT16、BFLOAT16	ND
weight	输入	表示可学习过滤器的4D张量，对应公式中的`weight`。数据类型、数据格式与入参`x`保持一致。shape为[outC, K_H, K_W, inC/groups]。	FLOAT32、FLOAT16、BFLOAT16	ND
offsets	输入	表示x-y坐标偏移和掩码的4D张量。对应公式中的`offset`。数据类型、数据格式与入参`x`保持一致。当`modulated`为true时，shape为[N, outH, outW, 3 * deformable_groups * K_H * K_W]；当`modulated`为false时，shape为[N, outH, outW, 2 * deformable_groups * K_H * K_W]。	FLOAT32、FLOAT16、BFLOAT16	ND
bias	可选输入	表示过滤器输出附加偏置的1D张量，预留参数。数据类型与入参`x`一致。不需要时为空指针，存在时shape为[outC]。	FLOAT32、FLOAT16、BFLOAT16	ND
kernel_size	可选属性	表示卷积核大小，对应公式中的`K_H`、`K_W`。size为2(K_H, K_W)，各元素均大于零，K_H * K_W不能超过2048，K_H * K_W * inC/groups不能超过65535。	ListInt	-
stride	可选属性	表示每个输入维度的滑动窗口步长，对应公式中的`stride`。size为4，各元素均大于零，维度顺序根据`x`的数据格式解释。N维和C维必须设置为1。	ListInt	-
padding	可选属性	表示要添加到输入每侧（顶部、底部、左侧、右侧）的像素数，对应公式中的`padding`。size为4。	FLOAT32	-
dilation	属性	表示输入每个维度的膨胀系数，对应公式中的`dilation`。size为4，各元素均大于零，维度顺序根据x的数据格式解释。N维和C维必须设置为1。 默认值为{1, 1, 1, 1}。	ListInt	-
groups	属性	表示从输入通道到输出通道的分组连接数。inC和outC需都可被`groups`数整除，`groups`数大于零。 默认值为1。	INT32	-
deformable_groups	属性	表示可变形组分区的数量。inC需可被`deformable_groups`数整除，`deformable_groups`的值大于零。 默认值为1。	INT32	-
modulated	属性	表示`offset`中是否包含掩码。若为true，`offset`中包含掩码；若为false，则不包含。 默认值为true，且当前仅支持配置为true。	BOOL	-
out	输出	表示计算的输出张量。对应公式中的`out`。数据类型、数据格式与`x`保持一致。shape为[N, outH, outC, outW]。	FLOAT32、FLOAT16、BFLOAT16	ND
deform_out	输出	表示计算的输出张量。对应公式中的`deformOut`。数据类型、数据格式与`x`保持一致。shape为[N, outH * K_H, outW * K_W, inC]。	FLOAT32、FLOAT16、BFLOAT16	ND

约束说明

无

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_deformable_conv2d	通过aclnnDeformableConv2d接口方式调用DeformableConv2d算子。