Skip to content

Commit d06b352

Browse files
authored
Add ops warpper (open-mmlab#3457)
* Add ops warpper * Reformat * disable yapf * Add comments
1 parent c342781 commit d06b352

File tree

2 files changed

+33
-1
lines changed

2 files changed

+33
-1
lines changed

.dev_scripts/linter.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
1-
yapf -r -i --style mmdet/ configs/ tests/ tools/
1+
yapf -r -i mmdet/ configs/ tests/ tools/
22
isort -rc mmdet/ configs/ tests/ tools/
33
flake8 .

mmdet/ops/__init__.py

+32
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
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

Comments
 (0)