From: Dmitry Kurtaev Date: Mon, 6 May 2019 06:13:47 +0000 (+0300) Subject: Modify paddings computation for SAME pad mode X-Git-Tag: accepted/tizen/6.0/unified/20201030.111113~1^2~249^2~5^2 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=471b83ccd5e4ea86327774ce39d9607cf78010e0;p=platform%2Fupstream%2Fopencv.git Modify paddings computation for SAME pad mode --- diff --git a/modules/dnn/src/layers/convolution_layer.cpp b/modules/dnn/src/layers/convolution_layer.cpp index 3b298e6..cfb93bb 100644 --- a/modules/dnn/src/layers/convolution_layer.cpp +++ b/modules/dnn/src/layers/convolution_layer.cpp @@ -125,7 +125,7 @@ public: inpShape.push_back(inputs[0].size[i]); outShape.push_back(outputs[0].size[i]); } - getConvPoolPaddings(inpShape, outShape, kernel_size, strides, padMode, dilations, pads_begin, pads_end); + getConvPoolPaddings(inpShape, kernel_size, strides, padMode, pads_begin, pads_end); if (pads_begin.size() == 2) { for (int i = 0; i < pads_begin.size(); i++) { if (pads_begin[i] != pads_end[i]) @@ -1257,7 +1257,7 @@ public: inpShape.push_back(inputs[0].size[i]); outShape.push_back(outputs[0].size[i]); } - getConvPoolPaddings(outShape, inpShape, kernel_size, strides, padMode, dilations, pads_begin, pads_end); + getConvPoolPaddings(outShape, kernel_size, strides, padMode, pads_begin, pads_end); if (pads_begin.size() == 2) { for (int i = 0; i < pads_begin.size(); i++) { if (pads_begin[i] != pads_end[i]) diff --git a/modules/dnn/src/layers/layers_common.cpp b/modules/dnn/src/layers/layers_common.cpp index 627f79c..29d863d 100644 --- a/modules/dnn/src/layers/layers_common.cpp +++ b/modules/dnn/src/layers/layers_common.cpp @@ -214,25 +214,25 @@ void getConvPoolOutParams(const std::vector& inp, const std::vector } } -void getConvPoolPaddings(const std::vector& inp, const std::vector& out, - const std::vector& kernel, const std::vector& strides, - const String &padMode, const std::vector& dilation, +void getConvPoolPaddings(const std::vector& inp, const std::vector& kernel, + const std::vector& strides, const String &padMode, std::vector& pads_begin, std::vector& pads_end) { - if (padMode == "VALID") + if (padMode == "SAME" || padMode == "VALID") { pads_begin.assign(kernel.size(), 0); pads_end.assign(kernel.size(), 0); } - else if (padMode == "SAME") + if (padMode == "SAME") { - CV_Assert_N(kernel.size() == dilation.size(), kernel.size() == strides.size(), - kernel.size() == inp.size(), kernel.size() == out.size()); - pads_begin.resize(kernel.size()); - pads_end.resize(kernel.size()); + CV_Assert_N(kernel.size() == strides.size(), kernel.size() == inp.size()); for (int i = 0; i < pads_begin.size(); i++) { - int pad = ((out[i] - 1) * strides[i] + dilation[i] * (kernel[i] - 1) + 1 - inp[i]) / 2; - pads_begin[i] = pads_end[i] = std::max(0, pad); + // There are test cases with stride > kernel. + if (strides[i] <= kernel[i]) + { + int pad = (kernel[i] - 1 - (inp[i] - 1 + strides[i]) % strides[i]) / 2; + pads_begin[i] = pads_end[i] = pad; + } } } } diff --git a/modules/dnn/src/layers/layers_common.hpp b/modules/dnn/src/layers/layers_common.hpp index fd1e430..26c1ce6 100644 --- a/modules/dnn/src/layers/layers_common.hpp +++ b/modules/dnn/src/layers/layers_common.hpp @@ -69,9 +69,8 @@ void getConvPoolOutParams(const std::vector& inp, const std::vector const std::vector& stride, const String &padMode, const std::vector& dilation, std::vector& out); - void getConvPoolPaddings(const std::vector& inp, const std::vector& out, - const std::vector& kernel, const std::vector& strides, - const String &padMode, const std::vector& dilation, + void getConvPoolPaddings(const std::vector& inp, const std::vector& kernel, + const std::vector& strides, const String &padMode, std::vector& pads_begin, std::vector& pads_end); } } diff --git a/modules/dnn/src/layers/pooling_layer.cpp b/modules/dnn/src/layers/pooling_layer.cpp index 78946b4..db5c849 100644 --- a/modules/dnn/src/layers/pooling_layer.cpp +++ b/modules/dnn/src/layers/pooling_layer.cpp @@ -143,7 +143,7 @@ public: kernel_size = std::vector(inp.begin(), inp.end()); } - getConvPoolPaddings(inp, out, kernel_size, strides, padMode, std::vector(kernel_size.size(), 1), pads_begin, pads_end); + getConvPoolPaddings(inp, kernel_size, strides, padMode, pads_begin, pads_end); if (pads_begin.size() == 2) { pad_t = pads_begin[0]; pad_l = pads_begin[1];