deformable_conv
deformable_conv
(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:
其中
和
分别为第k个位置的可学习偏移和调制标量。在deformable_conv_v1中
为1.
具体细节可以参考论文: 和 <<Deformable Convolutional Networks>> 。
- 输入:
- input 形状:
卷积核形状:
offset 形状:
mask 形状:
- 输出:
- 输出形状:
其中
- 参数:
返回:可变形卷积输出的4-D Tensor,数据类型为float32或float64。
返回类型:Variable
抛出异常:ValueError – 如果input, filter_size, stride, padding和groups的大小不匹配。
代码示例