"vscode:/vscode.git/clone" did not exist on "644bc76826fbd42f5100487671d6ef5a29bacbfd"
Commit d8c72d8b authored by Kai Chen's avatar Kai Chen
Browse files

minor fix

parent 1bd3f3f1
from .dcn import (DeformConv, DeformRoIPooling, ModulatedDeformRoIPoolingPack, from .dcn import (DeformConv, DeformRoIPooling, DeformRoIPoolingPack,
ModulatedDeformConv, ModulatedDeformConvPack, deform_conv, ModulatedDeformRoIPoolingPack, ModulatedDeformConv,
modulated_deform_conv, deform_roi_pooling) ModulatedDeformConvPack, deform_conv, modulated_deform_conv,
deform_roi_pooling)
from .nms import nms, soft_nms from .nms import nms, soft_nms
from .roi_align import RoIAlign, roi_align from .roi_align import RoIAlign, roi_align
from .roi_pool import RoIPool, roi_pool from .roi_pool import RoIPool, roi_pool
__all__ = [ __all__ = [
'nms', 'soft_nms', 'RoIAlign', 'roi_align', 'RoIPool', 'roi_pool', 'nms', 'soft_nms', 'RoIAlign', 'roi_align', 'RoIPool', 'roi_pool',
'DeformConv', 'DeformRoIPooling', 'ModulatedDeformRoIPoolingPack', 'DeformConv', 'DeformRoIPooling', 'DeformRoIPoolingPack',
'ModulatedDeformConv', 'ModulatedDeformConvPack', 'deform_conv', 'ModulatedDeformRoIPoolingPack', 'ModulatedDeformConv',
'modulated_deform_conv', 'deform_roi_pooling' 'ModulatedDeformConvPack', 'deform_conv', 'modulated_deform_conv',
'deform_roi_pooling'
] ]
...@@ -17,8 +17,8 @@ class DeformConv(nn.Module): ...@@ -17,8 +17,8 @@ class DeformConv(nn.Module):
padding=0, padding=0,
dilation=1, dilation=1,
deformable_groups=1, deformable_groups=1,
bias=None): bias=False):
assert bias is None assert not bias
super(DeformConv, self).__init__() super(DeformConv, self).__init__()
self.in_channels = in_channels self.in_channels = in_channels
self.out_channels = out_channels self.out_channels = out_channels
......
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