forked from SampannaKahu/ScanBank
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
57 lines (52 loc) · 1.78 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
49
50
51
52
53
54
55
56
57
#!/usr/bin/python
import os
import sys
from setuptools import setup, Extension, find_packages
tf_include = '/'.join(sys.executable.split('/')[:-2]) + \
'/lib/python%d.%d/site-packages/tensorflow/include' % sys.version_info[:2]
extra_defs = []
if os.uname().sysname == 'Darwin':
extra_defs.append('-D_GLIBCXX_USE_CXX11_ABI=0')
else:
os.environ['CC'] = 'g++'
os.environ['CXX'] = 'g++'
requirements_file = os.path.join(
os.path.dirname(__file__),
'requirements.txt')
requirements = open(requirements_file).read().split('\n')
requirements = [r for r in requirements if not '-e' in r]
setup(
name='deepfigures-open',
version='0.0.1',
url='http://github.com/SampannaKahu/deepfigures-open',
packages=find_packages(),
setup_requires=['Cython'],
install_requires=requirements,
tests_require=[],
zip_safe=False,
test_suite='py.test',
entry_points='',
cffi_modules=['deepfigures/utils/stringmatch/stringmatch_builder.py:ffibuilder'],
ext_modules=[
Extension(
name='tensorboxresnet.utils.stitch_wrapper',
sources=[
'./vendor/tensorboxresnet/tensorboxresnet/utils/stitch_wrapper.pyx',
'./vendor/tensorboxresnet/tensorboxresnet/utils/stitch_rects.cpp',
'./vendor/tensorboxresnet/tensorboxresnet/utils/hungarian/hungarian.cpp'
],
language='c++',
extra_compile_args=[
'-std=c++11',
'-Itensorbox/utils',
'-I%s' % tf_include
] + extra_defs,
)
],
package_data={
'vendor.tensorboxresnet.tensorboxresnet': [
'logging.conf'
]
},
include_package_data=True
)