/
setup.py
30 lines (23 loc) · 850 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
from Cython.Build import cythonize
import numpy as np
from setuptools import setup, find_packages, Extension
# 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(
'pycocotools._mask',
sources=['common/maskApi.c', 'pycocotools/_mask.pyx'],
include_dirs = [np.get_include(), 'common'],
extra_compile_args=['-Wno-cpp', '-Wno-unused-function', '-std=c99'],
)
]
setup(
name ='pydatatool',
version ='0.1',
packages = find_packages(),
author = "Zhewei Xu",
author_email= "xzhewei@gmail.com",
description = "A data tool for CNN Training.",
license = "MIT",
ext_modules = cythonize(ext_modules)
)