mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix bev nms (#1266)

pull/1275/head
q.yao 2022-10-28 16:08:37 +08:00 committed by GitHub
parent a2d6323cf1
commit f949414034
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mmdeploy/mmcv/ops/__init__.py
View File

@ -6,9 +6,10 @@ from . import roi_align # noqa: F401,F403
from . import roi_align_rotated # noqa: F401,F403
from . import transformer # noqa: F401,F403
from .nms import ONNXNMSop, TRTBatchedNMSop
from .nms_rotated import ONNXNMSRotatedOp, TRTBatchedRotatedNMSop
from .nms_rotated import (ONNXNMSRotatedOp, TRTBatchedBEVNMSop,
TRTBatchedRotatedNMSop)
__all__ = [
'ONNXNMSop', 'TRTBatchedNMSop', 'ONNXNMSRotatedOp',
'TRTBatchedRotatedNMSop'
'TRTBatchedRotatedNMSop', 'TRTBatchedBEVNMSop'
]