There was an error while loading. Please reload this page.
Export script for 3d-pose-baseline.
frozen_graph_def = tf.graph_util.convert_variables_to_constants(sess,sess.graph.as_graph_def(),["linear_model/add_1"]) import tf2onnx from tf2onnx.optimizer.transpose_optimizer import TransposeOptimizer graph1 = tf.Graph() with graph1.as_default(): tf.import_graph_def(frozen_graph_def) onnx_graph = tf2onnx.tfonnx.process_tf_graph(graph1, input_names=["import/inputs/enc_in:0"], output_names=["import/linear_model/add_1:0"],opset=10) optimizer = TransposeOptimizer() opt_model_proto = optimizer.optimize(onnx_graph) model_proto = onnx_graph.make_model("3d-pose-baseline") with open("3d-pose-baseline.onnx", "wb") as f: f.write(model_proto.SerializeToString())