From 13d0bfcfd0a00dce971af8347dfd93b922b840f7 Mon Sep 17 00:00:00 2001
From: Guoxia Wang <mingzilaochongtu@gmail.com>
Date: Thu, 7 Jan 2021 18:58:24 +0800
Subject: [PATCH] fix onnx2ncnn adaptive pool conversion bug, remove no used
 weights node (#2570)

---
 src/layer/arm/pooling_arm.cpp | 2 +-
 tools/onnx/onnx2ncnn.cpp      | 7 +++++++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/layer/arm/pooling_arm.cpp b/src/layer/arm/pooling_arm.cpp
index 888531dbe..ea72e6b59 100644
--- a/src/layer/arm/pooling_arm.cpp
+++ b/src/layer/arm/pooling_arm.cpp
@@ -42,7 +42,7 @@ Pooling_arm::Pooling_arm()
     support_bf16_storage = true;
 }
 
-int Pooling_arm::create_pipeline(const Option& _opt)
+int Pooling_arm::create_pipeline(const Option& /*opt*/)
 {
     if (adaptive_pooling)
     {
diff --git a/tools/onnx/onnx2ncnn.cpp b/tools/onnx/onnx2ncnn.cpp
index 08941ab9b..3b8edc6f3 100644
--- a/tools/onnx/onnx2ncnn.cpp
+++ b/tools/onnx/onnx2ncnn.cpp
@@ -2270,6 +2270,13 @@ int main(int argc, char** argv)
                 node_reference[node.input(1)] -= 1;
             }
         }
+        else if (op == "adaptive_avg_pool2d" || op == "adaptive_max_pool2d")
+        {
+            if (node.input_size() >= 2)
+            {
+                node_reference[node.input(1)] -= 1;
+            }
+        }
     }
 
     //         for (auto a: node_reference)