-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathsetup.py
22 lines (18 loc) · 960 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext
from Cython.Build import cythonize
import numpy as np
ext_modules = [
Extension("gmm", ["gmm.pyx"], libraries=["vl", "m"], include_dirs=[np.get_include()], language='c'),
Extension("kmeans", ["kmeans.pyx"], libraries=["vl", "m"], include_dirs=[np.get_include()], language='c'),
Extension("fisher", ["fisher.pyx"], libraries=["vl", "m"], include_dirs=[np.get_include()], language='c'),
Extension("vlad", ["vlad.pyx"], libraries=["vl", "m"], include_dirs=[np.get_include()], language='c'),
]
setup(name = 'vl_wrap',
cmdclass = {"build_Ext": build_ext},
ext_modules = cythonize(ext_modules))
'''
Extension("gmm_wrap", ["gmm_wrap.pyx"], libraries=["vl", "m"], include_dirs=[np.get_include()], language='c'),
Extension("fisher_wrap", ["fisher_wrap.pyx"], libraries=["vl", "m"], include_dirs=[np.get_include()], language='c'),
'''