Diff of /setup.py [000000] .. [d34869]

Switch to side-by-side view

--- a
+++ b/setup.py
@@ -0,0 +1,74 @@
+from setuptools import setup, find_packages
+from setuptools.extension import Extension
+from Cython.Build import cythonize
+import numpy
+import os
+import imp
+
+VERSION = imp.load_source('version', os.path.join('.', 'darkflow', 'version.py'))
+VERSION = VERSION.__version__
+
+if os.name == 'nt':
+    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()]
+                  )
+    ]
+
+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",
+                  sources=["darkflow/cython_utils/nms.pyx"],
+                  libraries=["m"]  # Unix-like specific
+                  ),
+        Extension("darkflow.cython_utils.cy_yolo2_findboxes",
+                  sources=["darkflow/cython_utils/cy_yolo2_findboxes.pyx"],
+                  libraries=["m"]  # Unix-like specific
+                  ),
+        Extension("darkflow.cython_utils.cy_yolo_findboxes",
+                  sources=["darkflow/cython_utils/cy_yolo_findboxes.pyx"],
+                  libraries=["m"]  # Unix-like specific
+                  )
+    ]
+
+setup(
+    version=VERSION,
+    name='darkflow',
+    description='Darkflow',
+    license='GPLv3',
+    url='https://github.com/thtrieu/darkflow',
+    packages=find_packages(),
+    scripts=['flow'],
+    ext_modules=cythonize(ext_modules)
+)