From af8e70e2784e7ac87a035f4a5256ea072b4e8cba Mon Sep 17 00:00:00 2001 From: fuchen Date: Wed, 6 Mar 2019 18:19:13 +0800 Subject: [PATCH] fix blank name for onnx model, use output name instead --- mace/python/tools/converter_tool/onnx_converter.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mace/python/tools/converter_tool/onnx_converter.py b/mace/python/tools/converter_tool/onnx_converter.py index 6befa478..fff5241c 100644 --- a/mace/python/tools/converter_tool/onnx_converter.py +++ b/mace/python/tools/converter_tool/onnx_converter.py @@ -230,6 +230,8 @@ def onnx_dtype(dtype): class OnnxNode(object): def __init__(self, node): self.name = str(node.name) + if self.name=='': + self.name = str(node.output) self.op_type = str(node.op_type) self.domain = str(node.domain) self.attrs = dict([(attr.name, -- GitLab