add pt2tf tool

2020-09-23 09:09:49 +08:00
parent 7f7b7df65d
commit 18aefa4dd0
407 changed files with 16211 additions and 0 deletions
@@ -0,0 +1,7 @@
+import onnx
+
+from onnx_tf.backend import prepare
+
+onnx_model = onnx.load("input_path")  # load onnx model
+tf_rep = prepare(onnx_model)  # prepare tf representation
+tf_rep.export_graph("output_path")  # export the model