__init__.py 820 Bytes
Newer Older
Kai Chen's avatar
Kai Chen committed
1
2
3
4
from .dcn import (DeformConv, DeformConvPack, ModulatedDeformConv,
                  ModulatedDeformConvPack, DeformRoIPooling,
                  DeformRoIPoolingPack, ModulatedDeformRoIPoolingPack,
                  deform_conv, modulated_deform_conv, deform_roi_pooling)
Kai Chen's avatar
Kai Chen committed
5
6
7
from .nms import nms, soft_nms
from .roi_align import RoIAlign, roi_align
from .roi_pool import RoIPool, roi_pool
Cao Yuhang's avatar
Cao Yuhang committed
8
from .sigmoid_focal_loss import SigmoidFocalLoss, sigmoid_focal_loss
Kai Chen's avatar
Kai Chen committed
9

yhcao6's avatar
yhcao6 committed
10
11
__all__ = [
    'nms', 'soft_nms', 'RoIAlign', 'roi_align', 'RoIPool', 'roi_pool',
Kai Chen's avatar
Kai Chen committed
12
    'DeformConv', 'DeformConvPack', 'DeformRoIPooling', 'DeformRoIPoolingPack',
Kai Chen's avatar
Kai Chen committed
13
14
    'ModulatedDeformRoIPoolingPack', 'ModulatedDeformConv',
    'ModulatedDeformConvPack', 'deform_conv', 'modulated_deform_conv',
Cao Yuhang's avatar
Cao Yuhang committed
15
    'deform_roi_pooling', 'SigmoidFocalLoss', 'sigmoid_focal_loss'
yhcao6's avatar
yhcao6 committed
16
]