From f7cbcaa72bedd0c8dd627a8c6676e05aac79fbbf Mon Sep 17 00:00:00 2001 From: nihuini Date: Wed, 3 Mar 2021 17:21:45 +0800 Subject: [PATCH] fix onnx normalize expand ghost shape --- tools/onnx/onnx2ncnn.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tools/onnx/onnx2ncnn.cpp b/tools/onnx/onnx2ncnn.cpp index 5ae5e954d..f1f913c3d 100644 --- a/tools/onnx/onnx2ncnn.cpp +++ b/tools/onnx/onnx2ncnn.cpp @@ -1204,6 +1204,10 @@ static void fuse_normalize(onnx::GraphProto* mutable_graph, std::mapoutput(0)] -= 1; } node_reference[node3->output(0)] -= 1; + if (node3->input_size() == 2) + { + node_reference[node3->input(1)] -= 1; + } blob_names.erase(node->output(0)); blob_names.erase(node2->output(0));