forked from samson-wang/cython_bbox
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
49 lines (40 loc) · 1.29 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
48
# --------------------------------------------------------
# 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 = cythonize([
Extension(
name='cython_bbox',
sources=['src/cython_bbox.pyx'],
extra_compile_args = ['-O3', '-g0'],
include_dirs=[numpy_include]
)
])
setup(
name='cython_bbox',
ext_modules=ext_modules,
version = '0.1.3',
description = 'Standalone cython_bbox',
long_description=long_description,
long_description_content_type="text/markdown",
author = 'Samson Wang',
url = 'https://github.com/samson-wang/cython_bbox.git',
keywords = ['cython_bbox'],
install_requires=['numpy']
)