Fix global axis

pull/16223/head
Liubov Batanina 5 years ago
parent 543e0302d3
commit cf477f7e9f
  1. 3
      modules/dnn/src/layers/pooling_layer.cpp

@ -95,8 +95,6 @@ public:
else
CV_Error(Error::StsBadArg, "Unknown pooling type \"" + pool + "\"");
global_axis = params.has("global_axis") ? params.get<int>("global_axis") : -1;
getPoolingKernelParams(params, kernel_size, globalPooling, pads_begin, pads_end, strides, padMode);
if (kernel_size.size() == 2) {
kernel = Size(kernel_size[1], kernel_size[0]);
@ -126,6 +124,7 @@ public:
CV_Error(Error::StsBadArg, "Cannot determine pooling type");
setParamsFrom(params);
ceilMode = params.get<bool>("ceil_mode", true);
global_axis = params.get<int>("global_axis", -1);
spatialScale = params.get<float>("spatial_scale", 1);
avePoolPaddedArea = params.get<bool>("ave_pool_padded_area", true);
}

Loading…
Cancel
Save