Merge pull request #17060 from dkurt:dnn_align_ie_pool

5 years ago · f6de25b6cb
parent b3514a5708 870a775d7d
commit f6de25b6cb
1 changed files with 1 additions and 1 deletions
--- a/modules/dnn/src/layers/pooling_layer.cpp
+++ b/modules/dnn/src/layers/pooling_layer.cpp
@ -325,7 +325,7 @@ public:
            ieLayer.setRoundingType(ceilMode ?
                                    InferenceEngine::Builder::PoolingLayer::RoundingType::CEIL :
                                    InferenceEngine::Builder::PoolingLayer::RoundingType::FLOOR);
-            ieLayer.setExcludePad(type == AVE && padMode == "SAME");
+            ieLayer.setExcludePad(!avePoolPaddedArea);

            InferenceEngine::Builder::Layer l = ieLayer;
            if (!padMode.empty())