--- 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) +)