Files
insightface/detection/RetinaFace/rcnn/pycocotools/setup.py
2020-11-06 13:59:21 +08:00

19 lines
572 B
Python

from distutils.core import setup
from Cython.Build import cythonize
from distutils.extension import Extension
import numpy as np
# To compile and install locally run "python setup.py build_ext --inplace"
# To install library to Python site-packages run "python setup.py build_ext install"
ext_modules = [
Extension(
'_mask',
sources=['maskApi.c', '_mask.pyx'],
include_dirs=[np.get_include()],
extra_compile_args=['-Wno-cpp', '-Wno-unused-function', '-std=c99'],
)
]
setup(name='pycocotools', ext_modules=cythonize(ext_modules))