diff --git a/mmdet/ops/__init__.py b/mmdet/ops/__init__.py index 21c2f327d019f05fd5a32575bb0c4ad71fe39ef8..9576335a3063a1e05ccc69b78fe28c5a85c16acb 100644 --- a/mmdet/ops/__init__.py +++ b/mmdet/ops/__init__.py @@ -1,14 +1,14 @@ -from .dcn import (DeformConv, DeformRoIPooling, DeformRoIPoolingPack, - ModulatedDeformRoIPoolingPack, ModulatedDeformConv, - ModulatedDeformConvPack, deform_conv, modulated_deform_conv, - deform_roi_pooling) +from .dcn import (DeformConv, DeformConvPack, ModulatedDeformConv, + ModulatedDeformConvPack, DeformRoIPooling, + DeformRoIPoolingPack, ModulatedDeformRoIPoolingPack, + deform_conv, modulated_deform_conv, deform_roi_pooling) from .nms import nms, soft_nms from .roi_align import RoIAlign, roi_align from .roi_pool import RoIPool, roi_pool __all__ = [ 'nms', 'soft_nms', 'RoIAlign', 'roi_align', 'RoIPool', 'roi_pool', - 'DeformConv', 'DeformRoIPooling', 'DeformRoIPoolingPack', + 'DeformConv', 'DeformConvPack', 'DeformRoIPooling', 'DeformRoIPoolingPack', 'ModulatedDeformRoIPoolingPack', 'ModulatedDeformConv', 'ModulatedDeformConvPack', 'deform_conv', 'modulated_deform_conv', 'deform_roi_pooling'