Added support for macOS

thtrieu · Apr 25, 2017 · 09839e0 · 09839e0
1 parent ee6fd2c
commit 09839e0
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -27,6 +27,25 @@
         )
     ]
 
+elif os.name =='posix' :
+    ext_modules=[
+        Extension("darkflow.cython_utils.nms",
+            sources=["darkflow/cython_utils/nms.pyx"],
+            libraries=["m"], # Unix-like specific
+            include_dirs=[numpy.get_include()]
+        ),        
+        Extension("darkflow.cython_utils.cy_yolo2_findboxes",
+            sources=["darkflow/cython_utils/cy_yolo2_findboxes.pyx"],
+            libraries=["m"], # Unix-like specific
+            include_dirs=[numpy.get_include()]
+        ),
+        Extension("darkflow.cython_utils.cy_yolo_findboxes",
+            sources=["darkflow/cython_utils/cy_yolo_findboxes.pyx"],
+            libraries=["m"], # Unix-like specific
+            include_dirs=[numpy.get_include()]
+        )
+    ]
+
 else :
     ext_modules=[
         Extension("darkflow.cython_utils.nms",