Files
ascend-tools/pt2pb/onnx-tensorflow/example/onnx_to_tf.py
T
2020-10-14 08:55:07 +08:00

8 lines
219 B
Python

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