Skip to content

Commit e421e83

Browse files
authored
port nms extension from maskrcnn-benchmark (open-mmlab#469)
* port nms extension from maskrcnn-benchmark * fix linting error
1 parent ee7e679 commit e421e83

File tree

14 files changed

+276
-345
lines changed

14 files changed

+276
-345
lines changed

.gitignore

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ venv.bak/
104104
.mypy_cache/
105105

106106
# cython generated cpp
107-
mmdet/ops/nms/*.cpp
107+
mmdet/ops/nms/src/soft_nms_cpu.cpp
108108
mmdet/version.py
109109
data
110110
.vscode

compile.sh

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,10 @@ $PYTHON setup.py build_ext --inplace
1818

1919
echo "Building nms op..."
2020
cd ../nms
21-
make clean
22-
make PYTHON=${PYTHON}
21+
if [ -d "build" ]; then
22+
rm -r build
23+
fi
24+
$PYTHON setup.py build_ext --inplace
2325

2426
echo "Building dcn..."
2527
cd ../dcn

mmdet/ops/nms/.gitignore

Lines changed: 0 additions & 1 deletion
This file was deleted.

mmdet/ops/nms/Makefile

Lines changed: 0 additions & 8 deletions
This file was deleted.

mmdet/ops/nms/cpu_nms.pyx

Lines changed: 0 additions & 70 deletions
This file was deleted.

mmdet/ops/nms/gpu_nms.hpp

Lines changed: 0 additions & 3 deletions
This file was deleted.

mmdet/ops/nms/gpu_nms.pyx

Lines changed: 0 additions & 45 deletions
This file was deleted.

mmdet/ops/nms/nms_kernel.cu

Lines changed: 0 additions & 188 deletions
This file was deleted.

0 commit comments

Comments
 (0)