（beta）torch_npu.contrib.DCNv2

[object Object]

产品支持情况

[object Object]undefined

功能说明

应用基于NPU的调制可变形2D卷积操作。ModulationDeformConv的实现主要是基于mmcv的实现进行设计和重构。

函数原型

参数说明

• in_channels（[object Object]）：输入图像的channel数量。
• out_channels（[object Object]）：卷积输出channel的数量。
• kernel_size（[object Object], [object Object]）：卷积核的尺寸。
• stride（[object Object], [object Object]）：卷积的stride，默认值为1。
• padding（[object Object] or [object Object]）：对输入图像两侧添加0填充，默认值为0。
• dilation（[object Object] or [object Object]）：卷积核元素间距，默认值为1。
• groups（[object Object]）：输入通道和输出通道的组数量，默认值为1。
• deformable_groups（[object Object]）：可变形分区数量。
• bias（[object Object]）：设置为True将对output添加bias，默认值为False。
• pack（[object Object]）：设置为True则将在模型中添加conv_offset和mask，默认值为True。

约束说明

ModulationDeformConv仅实现fp32数据类型下的操作。注意，conv_offset中的权重和偏置必须初始化为0。

调用示例