add pt2tf tool

This commit is contained in:
zxros10
2020-09-23 09:09:49 +08:00
parent 7f7b7df65d
commit 18aefa4dd0
407 changed files with 16211 additions and 0 deletions
+24
View File
@@ -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()