|
| 1 | +# This file is added for back-compatibility. Thus, downstream codebase |
| 2 | +# could still use and import mmdet.ops. |
| 3 | + |
| 4 | +# yapf: disable |
| 5 | +from mmcv.ops import (ContextBlock, Conv2d, ConvTranspose2d, ConvWS2d, |
| 6 | + CornerPool, DeformConv, DeformConvPack, DeformRoIPooling, |
| 7 | + DeformRoIPoolingPack, GeneralizedAttention, Linear, |
| 8 | + MaskedConv2d, MaxPool2d, ModulatedDeformConv, |
| 9 | + ModulatedDeformConvPack, ModulatedDeformRoIPoolingPack, |
| 10 | + NonLocal2D, RoIAlign, RoIPool, SAConv2d, |
| 11 | + SigmoidFocalLoss, SimpleRoIAlign, batched_nms, |
| 12 | + build_plugin_layer, conv_ws_2d, deform_conv, |
| 13 | + deform_roi_pooling, get_compiler_version, |
| 14 | + get_compiling_cuda_version, modulated_deform_conv, nms, |
| 15 | + nms_match, point_sample, rel_roi_point_to_rel_img_point, |
| 16 | + roi_align, roi_pool, sigmoid_focal_loss, soft_nms) |
| 17 | + |
| 18 | +# yapf: enable |
| 19 | + |
| 20 | +__all__ = [ |
| 21 | + 'nms', 'soft_nms', 'RoIAlign', 'roi_align', 'RoIPool', 'roi_pool', |
| 22 | + 'DeformConv', 'DeformConvPack', 'DeformRoIPooling', 'DeformRoIPoolingPack', |
| 23 | + 'ModulatedDeformRoIPoolingPack', 'ModulatedDeformConv', |
| 24 | + 'ModulatedDeformConvPack', 'deform_conv', 'modulated_deform_conv', |
| 25 | + 'deform_roi_pooling', 'SigmoidFocalLoss', 'sigmoid_focal_loss', |
| 26 | + 'MaskedConv2d', 'ContextBlock', 'GeneralizedAttention', 'NonLocal2D', |
| 27 | + 'get_compiler_version', 'get_compiling_cuda_version', 'ConvWS2d', |
| 28 | + 'conv_ws_2d', 'build_plugin_layer', 'batched_nms', 'Conv2d', |
| 29 | + 'ConvTranspose2d', 'MaxPool2d', 'Linear', 'nms_match', 'CornerPool', |
| 30 | + 'point_sample', 'rel_roi_point_to_rel_img_point', 'SimpleRoIAlign', |
| 31 | + 'SAConv2d' |
| 32 | +] |
0 commit comments