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