mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-03 14:50:47 +08:00
24 lines
538 B
Python
Executable File
24 lines
538 B
Python
Executable File
from distutils.core import setup
|
|
from distutils.extension import Extension
|
|
from Cython.Build import cythonize
|
|
import numpy as np
|
|
|
|
|
|
def numpy_include():
|
|
try:
|
|
numpy_include = np.get_include()
|
|
except AttributeError:
|
|
numpy_include = np.get_numpy_include()
|
|
return numpy_include
|
|
|
|
ext_modules = [
|
|
Extension('eval_metrics_cy',
|
|
['eval_metrics_cy.pyx'],
|
|
include_dirs=[numpy_include()],
|
|
)
|
|
]
|
|
|
|
setup(
|
|
name='Cython-based reid evaluation code',
|
|
ext_modules=cythonize(ext_modules)
|
|
) |