Commit fada959b authored by Liubov Batanina's avatar Liubov Batanina

Fix comment

parent d825caf1
...@@ -155,12 +155,7 @@ void getPoolingKernelParams(const LayerParams &params, std::vector<size_t>& kern ...@@ -155,12 +155,7 @@ void getPoolingKernelParams(const LayerParams &params, std::vector<size_t>& kern
globalPooling[2] = params.get<bool>("global_pooling_w", is_global); globalPooling[2] = params.get<bool>("global_pooling_w", is_global);
is_global = globalPooling[0] || globalPooling[1] || globalPooling[2]; is_global = globalPooling[0] || globalPooling[1] || globalPooling[2];
if (!is_global) if (is_global)
{
util::getKernelSize(params, kernel);
util::getStrideAndPadding(params, pads_begin, pads_end, strides, padMode, kernel.size());
}
else
{ {
util::getStrideAndPadding(params, pads_begin, pads_end, strides, padMode); util::getStrideAndPadding(params, pads_begin, pads_end, strides, padMode);
if ((globalPooling[0] && params.has("kernel_d")) || if ((globalPooling[0] && params.has("kernel_d")) ||
...@@ -184,6 +179,11 @@ void getPoolingKernelParams(const LayerParams &params, std::vector<size_t>& kern ...@@ -184,6 +179,11 @@ void getPoolingKernelParams(const LayerParams &params, std::vector<size_t>& kern
CV_Error(cv::Error::StsBadArg, "In global_pooling mode, strides must be = 1"); CV_Error(cv::Error::StsBadArg, "In global_pooling mode, strides must be = 1");
} }
} }
else
{
util::getKernelSize(params, kernel);
util::getStrideAndPadding(params, pads_begin, pads_end, strides, padMode, kernel.size());
}
} }
void getConvolutionKernelParams(const LayerParams &params, std::vector<size_t>& kernel, std::vector<size_t>& pads_begin, void getConvolutionKernelParams(const LayerParams &params, std::vector<size_t>& kernel, std::vector<size_t>& pads_begin,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment