From 2d2144639f6fa81dd39896926782e8c80761a703 Mon Sep 17 00:00:00 2001
From: thnkinbtfly <tbvj5914@gmail.com>
Date: Mon, 14 Dec 2020 05:31:10 +0000
Subject: [PATCH] support darknet-style avgpool (global avgpool)

---
 src/DarknetParser.cpp | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/DarknetParser.cpp b/src/DarknetParser.cpp
index d8c186a..891c97c 100644
--- a/src/DarknetParser.cpp
+++ b/src/DarknetParser.cpp
@@ -134,7 +134,11 @@ namespace tk { namespace dnn {
                     f.padding_x, f.padding_y, tk::dnn::POOLING_MAX));
 
         } else if(f.type == "avgpool") {
-            netLayers.push_back(new tk::dnn::Pooling(net, f.size_x, f.size_y, f.stride_x, f.stride_y, 
+            auto output_dim = net->getOutputDim();
+            int stride = 1;
+            assert(f.padding_x == 0 && f.padding_y == 0);
+
+            netLayers.push_back(new tk::dnn::Pooling(net, output_dim.h, output_dim.w, stride, stride,
                 f.padding_x, f.padding_y, tk::dnn::POOLING_AVERAGE));
 
         } else if(f.type == "shortcut") {