-
Notifications
You must be signed in to change notification settings - Fork 23
/
setup.py
67 lines (62 loc) · 2 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
58
59
60
61
62
63
64
65
66
67
from setuptools import setup, find_packages
from os import path
import re
package_name="simple_onnx_processing_tools"
root_dir = path.abspath(path.dirname(__file__))
with open("README.md") as f:
long_description = f.read()
with open(path.join(root_dir, package_name, '__init__.py')) as f:
init_text = f.read()
version = re.search(r'__version__\s*=\s*[\'\"](.+?)[\'\"]', init_text).group(1)
setup(
name=package_name,
version=version,
description=\
"A set of simple tools for splitting, merging, OP deletion, "+
"size compression, rewriting attributes and constants, "+
"OP generation, change opset, change to the specified input order, "+
"addition of OP, RGB to BGR conversion, change batch size, "+
"batch rename of OP, and JSON convertion for ONNX models.",
long_description=long_description,
long_description_content_type="text/markdown",
author="Katsuya Hyodo",
author_email="[email protected]",
url="https://github.com/PINTO0309/simple-onnx-processing-tools",
license="MIT License",
packages=find_packages(),
platforms=["linux", "unix"],
python_requires=">=3.6",
install_requires = [
'snc4onnx >= 1.0.12',
'sne4onnx >= 1.0.11',
'snd4onnx >= 1.1.6',
'scs4onnx >= 1.0.18',
'sog4onnx >= 1.0.16',
'sam4onnx >= 1.0.14',
'soc4onnx >= 1.0.2',
'scc4onnx >= 1.0.5',
'sna4onnx >= 1.0.6',
'sbi4onnx >= 1.0.5',
'sor4onnx >= 1.0.5',
'sit4onnx >= 1.0.7',
'onnx2json >= 2.0.4',
'json2onnx >= 2.0.3',
'sed4onnx >= 1.0.5',
'soa4onnx >= 1.0.4',
'sod4onnx >= 1.0.0',
'ssi4onnx >= 1.0.2',
'ssc4onnx >= 1.0.5',
'sio4onnx >= 1.0.2',
'svs4onnx >= 1.0.0',
'onnx2tf >= 1.20.1',
'sng4onnx >= 1.0.2',
'sde4onnx >= 1.0.0',
'spo4onnx >= 1.0.4',
],
extras_require = {
'full': [
'onnx-simplifier',
'rich',
],
},
)