重命名 pt2tf 为 pt2pb

pt2pb/onnx-tensorflow/onnx_tf/cli.py

@@ -0,0 +1,24 @@
+import argparse
+import sys
+
+import onnx_tf.converter
+
+
+def main():
+  args = sys.argv[1:]
+  parser = argparse.ArgumentParser(
+      description="ONNX-Tensorflow Command Line Interface")
+  parser.add_argument(
+      "command",
+      choices=["convert"],
+      help="Available commands.")
+
+  if len(args) == 0:
+    parser.parse_args(["-h"])
+  cli_tool = parser.parse_args([args[0]])
+  if cli_tool.command == "convert":
+    return onnx_tf.converter.main(args[1:])
+
+
+if __name__ == '__main__':
+  main()