| from setuptools import setup, Extension | |
| import numpy | |
| import os | |
| # Get absolute paths to OpenCV include and lib | |
| BASE_DIR = os.path.abspath(os.path.dirname(__file__)) | |
| OPENCV_DIR = os.path.join(BASE_DIR, 'opencv-install') | |
| INCLUDE_DIRS = [ | |
| numpy.get_include(), | |
| os.path.join(OPENCV_DIR, 'include'), | |
| os.path.join(OPENCV_DIR, 'include', 'opencv4') | |
| ] | |
| LIBRARY_DIRS = [os.path.join(OPENCV_DIR, 'lib')] | |
| # Define the extension | |
| _stasm = Extension( | |
| '_stasm', | |
| sources=[ | |
| 'src/MOD_1/facedet.cpp', # adjust this to actual source files | |
| 'src/MOD_1/your_other_file.cpp' | |
| ], | |
| include_dirs=INCLUDE_DIRS, | |
| library_dirs=LIBRARY_DIRS, | |
| libraries=[ | |
| 'opencv_core', | |
| 'opencv_imgproc', | |
| 'opencv_highgui', | |
| 'opencv_imgcodecs' | |
| ], | |
| extra_compile_args=['-std=c++11'] | |
| ) | |
| setup( | |
| name='stasm', | |
| version='2.0.2', | |
| packages=['stasm', 'stasm.data'], | |
| ext_modules=[_stasm], | |
| install_requires=[ | |
| 'numpy', | |
| 'docopt' | |
| ] | |
| ) |