From 298a5baa491c7083c092f76501ea795df9921862 Mon Sep 17 00:00:00 2001 From: zjhellofss Date: Sat, 21 Oct 2023 15:18:53 +0800 Subject: [PATCH] undo base pooling --- source/layer/details/maxpooling.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/source/layer/details/maxpooling.cpp b/source/layer/details/maxpooling.cpp index 27cf1b7..6c062e7 100644 --- a/source/layer/details/maxpooling.cpp +++ b/source/layer/details/maxpooling.cpp @@ -118,12 +118,12 @@ StatusCode MaxPoolingLayer::Forward(const std::vector::lowest(); for (uint32_t w = 0; w < pooling_w; ++w) { - const float* col_ptr = input_channel.colptr(c + w - padding_w_); + const float* col_ptr = input_channel.colptr(c + w - padding_w_) + r; for (uint32_t h = 0; h < pooling_h; ++h) { float current_value = 0.f; if ((h + r >= padding_h_ && w + c >= padding_w_) && (h + r < input_h + padding_h_ && w + c < input_w + padding_w_)) { - current_value = *(col_ptr + r + h - padding_h_); + current_value = *(col_ptr + h - padding_h_); } else { current_value = std::numeric_limits::lowest(); }