From 870a775d7d891c82278723485d5719fd2f69748f Mon Sep 17 00:00:00 2001 From: Dmitry Kurtaev Date: Tue, 14 Apr 2020 13:54:54 +0300 Subject: [PATCH] Align DLIE and nGraph average pooling layers --- modules/dnn/src/layers/pooling_layer.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/dnn/src/layers/pooling_layer.cpp b/modules/dnn/src/layers/pooling_layer.cpp index 83f3df0522..cd16ec7f3b 100644 --- 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())