未验证 提交 75c9a63b 编写于 作者: J Jason 提交者: GitHub

Merge pull request #126 from mamingjie-China/develop

fix the bug and update the README
......@@ -9,7 +9,7 @@ X2Paddle在多个主流的CV模型上,测试过TensorFlow/Caffe/ONNX模型的
## 环境依赖
python >= 3.5
python == 2.7 | python >= 3.5
paddlepaddle >= 1.5.0
**按需安装以下依赖**
......
......@@ -785,6 +785,9 @@ class TFOpMapper(OpMapper):
start = self.graph.get_node(node.layer.input[0], copy=True)
limit = self.graph.get_node(node.layer.input[1], copy=True)
delta = self.graph.get_node(node.layer.input[2], copy=True)
self.add_omit_nodes(start.layer_name, node.layer_name)
self.add_omit_nodes(limit.layer_name, node.layer_name)
self.add_omit_nodes(delta.layer_name, node.layer_name)
if start.layer_type == "Const":
start = start.value
else:
......@@ -797,9 +800,6 @@ class TFOpMapper(OpMapper):
delta = delta.value
else:
delta = self.decoder.infer_tensor(delta)
self.add_omit_nodes(start.layer_name, node.layer_name)
self.add_omit_nodes(limit.layer_name, node.layer_name)
self.add_omit_nodes(delta.layer_name, node.layer_name)
inputs = {"start": start, "end": limit, "step": delta}
attr = {"dtype": string(node.dtype)}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册