Add cython util

Author: yukitsuji

cython_util/setup.py

@@ -103,6 +103,18 @@ def build_extensions(self):
         build_ext.build_extensions(self)
 
 ext_modules = [
+    Extension(
+        "nms",
+        ["nms.pyx"],
+        extra_compile_args={'gcc': ["-Wno-cpp", "-Wno-unused-function"]},
+        include_dirs = [numpy_include]
+    ),
+    Extension(
+        "proposal_for_train",
+        ["proposal_for_train.pyx"],
+        extra_compile_args={'gcc': ["-Wno-cpp", "-Wno-unused-function"]},
+        include_dirs = [numpy_include]
+    ),
     Extension(
         "bbox_overlap",
         ["bbox_overlap.pyx"],
@@ -121,18 +133,6 @@ def build_extensions(self):
         extra_compile_args={'gcc': ["-Wno-cpp", "-Wno-unused-function"]},
 	include_dirs = [numpy_include]
 	),
-    # Extension(
-    #     "utils.cython_nms",
-    #     ["utils/nms.pyx"],
-    #     extra_compile_args={'gcc': ["-Wno-cpp", "-Wno-unused-function"]},
-    #     include_dirs = [numpy_include]
-    # ),
-    # Extension(
-    #     "nms.cpu_nms",
-    #     ["nms/cpu_nms.pyx"],
-    #     extra_compile_args={'gcc': ["-Wno-cpp", "-Wno-unused-function"]},
-    #     include_dirs = [numpy_include]
-    # )
 ]
 
 # if CUDA: