Unverified Commit 9a3f1249 authored by Ming-Hsuan-Tu's avatar Ming-Hsuan-Tu Committed by GitHub
Browse files

[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: default avatarmaningsheng <maningsheng@sensetime.com>
parent bcf85026
......@@ -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)
......
......@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment