From 007a9e731a794ce66718db4aace5b3c99ff0ee36 Mon Sep 17 00:00:00 2001
From: GameGaoke <1210006776@qq.com>
Date: Fri, 28 Sep 2018 11:31:20 +0800
Subject: [PATCH] Update setup.py

fix  ImportError: No module named 'utils.cython_nms'
---
 faster_rcnn/setup.py | 6 ++++++
 1 file changed, 6 insertions(+)

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(