deformable_conv
API属性:声明式编程(静态图)专用API
( input, offset, mask, num_filters, filter_size, stride=1, padding=0, dilation=1, groups=None, deformable_groups=None, im2col_step=None, param_attr=None, bias_attr=None, modulated=True, name=None ) [源代码]
可变形卷积算子
deformable_conv op对输入4-D Tensor计算2-D可变形卷积。给定输入Tensor x,输出Tensor y,可变形卷积运算如下所示:
可形变卷积v2:
可形变卷积v1:
其中
和
为1.
具体细节可以参考论文: 和 <<Deformable Convolutional Networks>> 。
示例
输入:
input 形状:
卷积核形状:
offset 形状:
mask 形状:
输出:
输出形状:
其中
可变形卷积输出的4-D Tensor,数据类型为float32或float64。
ValueError – 如果input, filter_size, stride, padding和groups的大小不匹配。