__init__.py 607 Bytes
Newer Older
yhcao6's avatar
yhcao6 committed
1
2
3
from .dcn import (DeformConv, DeformRoIPooling, ModulatedDeformRoIPoolingPack,
                  ModulatedDeformConv, ModulatedDeformConvPack, deform_conv,
                  modulated_deform_conv, deform_roi_pooling)
Kai Chen's avatar
Kai Chen committed
4
5
6
from .nms import nms, soft_nms
from .roi_align import RoIAlign, roi_align
from .roi_pool import RoIPool, roi_pool
Kai Chen's avatar
Kai Chen committed
7

yhcao6's avatar
yhcao6 committed
8
9
10
11
12
13
__all__ = [
    'nms', 'soft_nms', 'RoIAlign', 'roi_align', 'RoIPool', 'roi_pool',
    'DeformConv', 'DeformRoIPooling', 'ModulatedDeformRoIPoolingPack',
    'ModulatedDeformConv', 'ModulatedDeformConvPack', 'deform_conv',
    'modulated_deform_conv', 'deform_roi_pooling'
]