Rewrite Soft NMS with pytorch extension (#2056)
* add soft_nms_cpu * remove cython dependency * merge soft_nms_cpu.cpp into nms_cpu.cpp * add docs * fixed typo * fixed typo * fixed typo
Showing
- .gitignore 0 additions, 2 deletions.gitignore
- .isort.cfg 1 addition, 1 deletion.isort.cfg
- mmdet/ops/nms/nms_wrapper.py 30 additions, 15 deletionsmmdet/ops/nms/nms_wrapper.py
- mmdet/ops/nms/src/nms_cpu.cpp 145 additions, 1 deletionmmdet/ops/nms/src/nms_cpu.cpp
- mmdet/ops/nms/src/soft_nms_cpu.pyx 0 additions, 127 deletionsmmdet/ops/nms/src/soft_nms_cpu.pyx
- requirements/build.txt 0 additions, 1 deletionrequirements/build.txt
- setup.py 1 addition, 27 deletionssetup.py
- tests/test_soft_nms.py 41 additions, 0 deletionstests/test_soft_nms.py
Loading
Please register or sign in to comment