[fix] Missing arguments when converting dcn to onnx (#624)

* fix issues when converting deformable convolution to onnx

* keep  and  for interface consistency

Co-authored-by: maningsheng <maningsheng@sensetime.com>
pull/644/head
Ming-Hsuan-Tu 2020-11-06 13:01:28 +08:00 committed by GitHub
parent bcf85026c3
commit 9a3f1249e5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 23 additions and 15 deletions

View File

@ -20,20 +20,29 @@ ext_module = ext_loader.load_ext('_ext', [
class DeformConv2dFunction(Function):
@staticmethod
def symbolic(g, input, offset, weight, stride, padding, dilation, groups,
deform_groups, bias, im2col_step):
def symbolic(g,
input,
offset,
weight,
stride,
padding,
dilation,
groups,
deform_groups,
bias=False,
im2col_step=32):
return g.op(
'MMCVDeformConv2d',
input,
offset,
weight,
stride=stride,
padding=padding,
dilation=dilation,
groups=groups,
deform_groups=deform_groups,
bias=bias,
im2col_step=im2col_step)
stride_i=stride,
padding_i=padding,
dilation_i=dilation,
groups_i=groups,
deform_groups_i=deform_groups,
bias_i=bias,
im2col_step_i=im2col_step)
@staticmethod
def forward(ctx,
@ -52,7 +61,6 @@ class DeformConv2dFunction(Function):
f'Expected 4D tensor as input, got {input.dim()}D tensor \
instead.')
assert bias is False, 'Only support bias is False.'
ctx.stride = _pair(stride)
ctx.padding = _pair(padding)
ctx.dilation = _pair(dilation)

View File

@ -27,11 +27,11 @@ class ModulatedDeformConv2dFunction(Function):
mask,
weight,
bias,
stride=stride,
padding=padding,
dilation=dilation,
groups=groups,
deform_groups=deform_groups)
stride_i=stride,
padding_i=padding,
dilation_i=dilation,
groups_i=groups,
deform_groups_i=deform_groups)
@staticmethod
def forward(ctx,