forked from samson-wang/cython_bbox
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathsetup.py
48 lines (39 loc) · 1.26 KB
/
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
# --------------------------------------------------------
# Fast R-CNN
# Copyright (c) 2015 Microsoft
# Licensed under The MIT License [see LICENSE for details]
# Written by Samson Wang
# --------------------------------------------------------
from __future__ import print_function
from Cython.Build import cythonize
from Cython.Distutils import build_ext
from setuptools import Extension
from setuptools import setup
import numpy as np
# Obtain the numpy include directory. This logic works across numpy versions.
try:
numpy_include = np.get_include()
except AttributeError:
numpy_include = np.get_numpy_include()
with open("README.md", "r") as fh:
long_description = fh.read()
ext_modules = [
Extension(
name='cython_bbox',
sources=['src/cython_bbox.pyx'],
extra_compile_args = ['-Wno-cpp' ],
include_dirs=[numpy_include]
)
]
setup(
name='cython_bbox',
ext_modules=cythonize(ext_modules),
version = '0.1.3',
description = 'Standalone cython_bbox',
long_description=long_description,
long_description_content_type="text/markdown",
author = 'Samson Wang',
author_email = '[email protected]',
url = 'https://github.com/samson-wang/cython_bbox.git',
keywords = ['cython_bbox']
)