昇腾社区首页
中文
注册

DeformableConv2D

功能

形变卷积。

输入

四个输入:

x:输入4Dtensor,格式为NCHW。

filter:权重tensor,格式为NCHW。

offsets:偏移量,4维tensor,格式为NCHW。

bias:可选,偏差,一维tensor,格式为ND。

输出

一个输出:

y:形变卷积输出tensor,格式为NCHW。

属性

auto_pad:可选,支持VALID、NOTSET。

dilations:4个整数的列表,指定用于扩张卷积的扩张率,H和W维度取值范围为[1, 255]。

groups:从输入通道到输出通道的阻塞连接数,输入通道和输出通道都必须被“groups”整除;数据类型为int32,必须设置为1。

pads:4个整数的列表,指定顶部、底部、左侧和右侧填充,取值范围为[0, 255]。

strides:4个整数的列表,指定沿高度H和宽度W的卷积步长。H和W维度取值范围为[1, 63],默认情况下,N和C尺寸设置为1。

data_format:string,表示输入数据format,默认是“NHWC”。

deformable_groups:分组卷积通道数,缺省为1。

modulated:bool,指定DeformableConv2D版本,true表示v2版本,false表示v1版本,当前只支持true。

约束

输入tensor,W维度取值范围为[1, 4096 / filter_width],H取值范围为[1, 100000 / filter_height]。

权重tensor,W维度取值范围为[1, 63],H取值范围为[1, 63]。

不支持atc工具--precision_mode=must_keep_origin_dtype参数时输入类型为float和float64。

支持的ONNX版本

Opset v9/v10/v11/v12/v13/v14/v15/v16