diff --git a/faster_rcnn/setup.py b/faster_rcnn/setup.py index 306aecf..75d85d7 100644 --- a/faster_rcnn/setup.py +++ b/faster_rcnn/setup.py @@ -154,6 +154,12 @@ def build_extensions(self): extra_compile_args={ 'gcc': ['-Wno-cpp', '-Wno-unused-function', '-std=c99']}, ), + Extension( + "utils.cython_nms", + sources = ["utils/nms.pyx"], + extra_compile_args={'gcc': []}, + include_dirs = [numpy_include], + ), ] setup(