-
Notifications
You must be signed in to change notification settings - Fork 73
/
setup.py
36 lines (31 loc) · 1.11 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
from setuptools import setup, find_packages
from os import path
import re
package_name="onnx2tf"
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=\
"Self-Created Tools to convert ONNX files (NCHW) to TensorFlow/TFLite/Keras format (NHWC). "+
"The purpose of this tool is to solve the massive Transpose extrapolation problem in onnx-tensorflow (onnx-tf).",
long_description=long_description,
long_description_content_type="text/markdown",
author="Katsuya Hyodo",
author_email="[email protected]",
url="https://github.com/PINTO0309/onnx2tf",
license="MIT License",
packages=find_packages(exclude=['test*','json_samples']),
platforms=["linux", "unix"],
python_requires=">=3.10",
entry_points={
'console_scripts': [
"onnx2tf=onnx2tf:main"
]
}
)