重命名 pt2tf 为 pt2pb

2020-10-14 08:55:07 +08:00
parent 324ab60a5d
commit 90ae190559
407 changed files with 0 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