make convert.py has same procedure with run_pretrained_models.py

Author: zhijxu-MS

tf2onnx/convert.py

@@ -14,7 +14,7 @@
 
 import tensorflow as tf
 
-from tf2onnx.graph import GraphUtil
+from tf2onnx import optimizer
 from tf2onnx.tfonnx import process_tf_graph, tf_optimize
 from tf2onnx import constants, loader, logging, utils
 
@@ -145,10 +145,10 @@ def main():
                              output_names=outputs,
                              inputs_as_nchw=args.inputs_as_nchw)
 
-    model_proto = g.make_model("converted from {}".format(model_path))
-
-    logger.info("")
-    model_proto = GraphUtil.optimize_model_proto(model_proto)
+    onnx_graph = optimizer.optimize_graph(g)
+    if not onnx_graph:
+        onnx_graph = g
+    model_proto = onnx_graph.make_model("converted from {}".format(model_path))
 
     # write onnx graph
     logger.info("")