add pt2tf tool

pt2tf/onnx-tensorflow/setup.py

@@ -0,0 +1,59 @@
+import os
+import setuptools
+import subprocess
+from textwrap import dedent
+
+TOP_DIR = os.path.realpath(os.path.dirname(__file__))
+SRC_DIR = os.path.join(TOP_DIR, 'onnx_tf')
+
+with open(os.path.join(TOP_DIR, 'VERSION_NUMBER')) as version_file:
+  version = version_file.read().strip()
+
+if os.getenv('TRAVIS'):
+  # On travis, we install from source, therefore no need to specify version.
+  onnx_dep = "onnx"
+else:
+  # For user, we install the onnx release known to work with our release.
+  with open(os.path.join(TOP_DIR, 'ONNX_VERSION_NUMBER')) as onnx_version_file:
+    onnx_version = onnx_version_file.read().strip()
+    onnx_dep = "onnx>=" + onnx_version
+
+try:
+  git_version = subprocess.check_output(['git', 'rev-parse', 'HEAD'],
+                                        cwd=TOP_DIR).decode('ascii').strip()
+except (OSError, subprocess.CalledProcessError):
+  git_version = None
+
+with open(os.path.join(SRC_DIR, 'version.py'), 'w') as f:
+  f.write(dedent('''\
+  # This file is generated by setup.py. DO NOT EDIT!
+  from __future__ import absolute_import
+  from __future__ import division
+  from __future__ import print_function
+  from __future__ import unicode_literals
+  version = '{}'
+  git_version = '{}'
+  '''.format(version, git_version)))
+
+setuptools.setup(
+    name='onnx-tf',
+    version=version,
+    description=
+    'Tensorflow backend for ONNX (Open Neural Network Exchange).',
+    install_requires=[onnx_dep, "PyYAML"],
+    entry_points={
+        "console_scripts": [
+            "onnx-tf=onnx_tf.cli:main",
+        ],
+    },
+    url='https://github.com/onnx/onnx-tensorflow/',
+    author='Arpith Jacob, Tian Jin, Gheorghe-Teodor Bercea, Wenhao Hu',
+    author_email='tian.jin1@ibm.com',
+    license='Apache License 2.0',
+    packages=setuptools.find_packages(),
+    zip_safe=False, 
+    classifiers=[
+         "Programming Language :: Python :: 2",
+         "Programming Language :: Python :: 3"
+    ]
+)