port nms extension from maskrcnn-benchmark (#469)
* port nms extension from maskrcnn-benchmark * fix linting error
Showing
- .gitignore 1 addition, 1 deletion.gitignore
- compile.sh 4 additions, 2 deletionscompile.sh
- mmdet/ops/nms/.gitignore 0 additions, 1 deletionmmdet/ops/nms/.gitignore
- mmdet/ops/nms/Makefile 0 additions, 8 deletionsmmdet/ops/nms/Makefile
- mmdet/ops/nms/cpu_nms.pyx 0 additions, 70 deletionsmmdet/ops/nms/cpu_nms.pyx
- mmdet/ops/nms/gpu_nms.hpp 0 additions, 3 deletionsmmdet/ops/nms/gpu_nms.hpp
- mmdet/ops/nms/gpu_nms.pyx 0 additions, 45 deletionsmmdet/ops/nms/gpu_nms.pyx
- mmdet/ops/nms/nms_kernel.cu 0 additions, 188 deletionsmmdet/ops/nms/nms_kernel.cu
- mmdet/ops/nms/nms_wrapper.py 34 additions, 19 deletionsmmdet/ops/nms/nms_wrapper.py
- mmdet/ops/nms/setup.py 17 additions, 7 deletionsmmdet/ops/nms/setup.py
- mmdet/ops/nms/src/nms_cpu.cpp 71 additions, 0 deletionsmmdet/ops/nms/src/nms_cpu.cpp
- mmdet/ops/nms/src/nms_cuda.cpp 17 additions, 0 deletionsmmdet/ops/nms/src/nms_cuda.cpp
- mmdet/ops/nms/src/nms_kernel.cu 131 additions, 0 deletionsmmdet/ops/nms/src/nms_kernel.cu
- mmdet/ops/nms/src/soft_nms_cpu.pyx 1 addition, 1 deletionmmdet/ops/nms/src/soft_nms_cpu.pyx
Loading
Please register or sign in to comment