未验证 提交 e3de1fa3 编写于 作者: W WJJ1995 提交者: GitHub

Support for pose_landmarks (#716)

* fixed split bug

* fixed concat bug

* fixed node_name bugs

* fixed node_name bugs

* fixed CI bugs
上级 fab94829
...@@ -583,6 +583,8 @@ class ONNXDecoder(object): ...@@ -583,6 +583,8 @@ class ONNXDecoder(object):
item.name = self.make_variable_name(item.name) item.name = self.make_variable_name(item.name)
for node in graph.node: for node in graph.node:
node.name = node.output[0] node.name = node.output[0]
if ":" in node.name and len(node.output) > 1:
node.name = node.name.split(':')[0]
node.name = self.make_variable_name(node.name) node.name = self.make_variable_name(node.name)
for i in range(len(node.input)): for i in range(len(node.input)):
if node.input[i] == '': if node.input[i] == '':
......
...@@ -1366,8 +1366,6 @@ class OpSet9(): ...@@ -1366,8 +1366,6 @@ class OpSet9():
val_x = self.graph.get_input_node(node, idx=0, copy=True) val_x = self.graph.get_input_node(node, idx=0, copy=True)
paddle_op = 'split' paddle_op = 'split'
split = node.get_attr('split') split = node.get_attr('split')
if split is None:
split = len(node.outputs)
axis = node.get_attr('axis', 0) axis = node.get_attr('axis', 0)
if split is None: if split is None:
split_num = len(node.layer.output) split_num = len(node.layer.output)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册