setup.py

# Third party imports
import numpy as np
from Cython.Build import cythonize
from setuptools import Extension, find_packages, setup

ext_modules = [
    Extension(
        'util.nms',
        sources=['util/external/nms.pyx'],
        include_dirs=[np.get_include()],
        extra_compile_args=['-Wno-cpp', '-Wno-unused-function', '-std=c99'],
    )
]

setup(
    name='util',
    ext_modules=cythonize(ext_modules)
)