From 394814e82c93ba8784df787c685e68aad16f0fc8 Mon Sep 17 00:00:00 2001
From: jiangjiajun <jiangjiajun@baidu.com>
Date: Mon, 5 Aug 2019 10:17:38 +0800
Subject: [PATCH] little modify

---
 x2paddle/core/util.py          |  6 ------
 x2paddle/decoder/tf_decoder.py | 35 ++++++++--------------------------
 2 files changed, 8 insertions(+), 33 deletions(-)

diff --git a/x2paddle/core/util.py b/x2paddle/core/util.py
index cc58748..45048b8 100644
--- a/x2paddle/core/util.py
+++ b/x2paddle/core/util.py
@@ -38,9 +38,3 @@ def run_net(param_dir="./"):
                        param_dir,
                        fluid.default_main_program(),
                        predicate=if_exist)
-
-    fluid.io.save_inference_model(dirname='inference_model',
-                                  feeded_var_names=[i.name for i in inputs],
-                                  target_vars=outputs,
-                                  executor=exe,
-                                  params_filename="__params__")
diff --git a/x2paddle/decoder/tf_decoder.py b/x2paddle/decoder/tf_decoder.py
index 014250a..cbf4b84 100644
--- a/x2paddle/decoder/tf_decoder.py
+++ b/x2paddle/decoder/tf_decoder.py
@@ -1,3 +1,4 @@
+#coding:utf-8
 #   Copyright (c) 2019  PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License"
@@ -181,24 +182,6 @@ class TFGraph(Graph):
 
             self.identity_map[node_name] = input_node.layer_name
 
-            #            node = self.get_node(node_name)
-            #            # Remind: Only 1 input for Identity node
-            #            input_node = self.get_node(node.inputs[0])
-            #
-            #            # remove identity node from graph
-            #            self.identity_map[node_name] = input_node.layer_name
-            #            idx = input_node.outputs.index(node_name)
-            #            del input_node.outputs[idx]
-            #
-            #            output_names = node.outputs
-            #            for output_name in output_names:
-            #                output_node = self.get_node(output_name)
-            #                idx = output_node.inputs.index(node_name)
-            #                output_node.inputs[idx] = input_node.layer_name
-            #
-            #            idx = self.topo_sort.index(node_name)
-            #            del self.topo_sort[idx]
-
             if node_name in self.output_nodes:
                 idx = self.output_nodes.index(node_name)
                 self.output_nodes[idx] = input_node.layer_name
@@ -227,10 +210,6 @@ class TFDecoder(object):
             self.sess.graph.as_default()
             tf.import_graph_def(graph_def, name='', input_map=input_map)
 
-
-#        for node in graph_def.node:
-#            print(node.name, node.op, node.input)
-
         self.sess.run(tf.global_variables_initializer())
 
         self.tf_graph = TFGraph(
@@ -264,13 +243,17 @@ class TFDecoder(object):
 
             if need_define_shape > 0:
                 if need_define_shape == 1:
-                    print(
-                        "\nUnknown shape for input tensor[tensor name: \"{}\"]".
-                        format(layer.name))
+                    print("无法获取到输入结点\"{}\"的shape".format(layer.name))
+                    print("Unknown shape for input tensor[tensor name: \"{}\"]".
+                          format(layer.name))
                 else:
+                    print(
+                        "输入结点\"{}\"的shape为{}，但我们现仅支持batch维为不定长，所以需要你重新设定shape".
+                        format(layer.name, shape))
                     print(
                         "\nShape[now is {}] for input tensor[tensor name: \"{}\"] not support yet"
                         .format(shape, layer.name))
+                print("需要你手动在下面输入对应这个输入结点的shape:)")
                 print(
                     "Use your keyboard type the shape of input tensor below :)")
 
@@ -293,13 +276,11 @@ class TFDecoder(object):
                                                     layer.name))
                 input_map["{}:0".format(layer.name)] = x2paddle_input
                 shape[shape.index(None)] = -1
-                #                self.input_example_data["x2paddle_{}".format(layer.name)] = numpy.random.random_sample(shape).astype(dtype)
                 self.input_info["x2paddle_{}".format(layer.name)] = (shape,
                                                                      dtype)
             else:
                 value = graph_node.layer.attr["shape"].shape
                 shape = [dim.size for dim in value.dim]
-                #                self.input_example_data[graph_node.layer_name] = numpy.random.random_sample(shape).astype(dtype)
                 self.input_info[graph_node.layer_name] = (shape, dtype)
 
         return input_map
-- 
GitLab