Skip to content
Snippets Groups Projects
Commit d8c72d8b authored by Kai Chen's avatar Kai Chen
Browse files

minor fix

parent 1bd3f3f1
No related branches found
No related tags found
No related merge requests found
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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment