Files
ascend-tools/pt2tf/onnx-tensorflow/example/onnx_to_tf.py
T
2020-09-23 09:09:49 +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