setup_nms_gpu.py

#!/usr/bin/env python

from distutils.core import setup
from setuptools import find_packages

from Cython.Distutils import build_ext
from distutils.extension import Extension
import numpy as np


description = """
Collection of Deep Learning Computer Vision Algorithms implemented in Chainer
"""

ext_modules = [
    Extension('time_axis_rcnn.model.time_segment_network.util.bbox._nms_gpu_post',
              ['time_axis_rcnn/model/time_segment_network/util/bbox/_nms_gpu_post.pyx']),
]
cmdclass = {'build_ext': build_ext}

install_requires = [
    'chainer==4.0',
    'Cython',
    'Pillow'
]

setup(
    name='time_axis_rcnn',
    version='0.5.1',
    packages=find_packages(),
    author='Yusuke Niitani',
    author_email='yuyuniitani@gmail.com',
    license='MIT',
    description=description,
    install_requires=install_requires,
    include_package_data=True,
    # for Cython
    ext_modules=ext_modules,
    cmdclass=cmdclass,
    include_dirs=[np.get_include()],
)