from distutils.core import setup
from distutils.extension import Extension

import numpy
from Cython.Build import cythonize

extensions = [
    Extension(
        "bbox",
        ["bbox.pyx"],
        extra_compile_args=[]
    ),
    Extension(
        "nms",
        ["nms.pyx"],
        extra_compile_args=[]
    )
]

setup(
    name="coco",
    ext_modules=cythonize(extensions),
    include_dirs=[numpy.get_include()]
)
