diff --git a/modules/dnn/src/layers/convolution_layer.cpp b/modules/dnn/src/layers/convolution_layer.cpp index f131c023ab..ef1dc8f29a 100644 --- a/modules/dnn/src/layers/convolution_layer.cpp +++ b/modules/dnn/src/layers/convolution_layer.cpp @@ -117,6 +117,9 @@ public: { kernel_size.assign(1, kernel_size[0]); strides.assign(1, strides[0]); + dilations.assign(1, dilations[0]); + pads_begin.assign(1, pads_begin[0]); + pads_end.assign(1, pads_end[0]); } CV_Assert(weightShape.dims() == kernel_size.size() + 2); for (int i = 0; i < kernel_size.size(); i++) { diff --git a/modules/dnn/src/layers/pooling_layer.cpp b/modules/dnn/src/layers/pooling_layer.cpp index dbe2332b89..834493966f 100644 --- a/modules/dnn/src/layers/pooling_layer.cpp +++ b/modules/dnn/src/layers/pooling_layer.cpp @@ -167,14 +167,13 @@ public: if (inputs[0].dims == 3) { - //Pool1D - kernel_size.erase(kernel_size.begin() + 1); - strides.erase(strides.begin() + 1); - pads_begin.erase(pads_begin.begin() + 1); - pads_end.erase(pads_end.begin() + 1); + // Pool1D + kernel_size.assign(1, kernel_size[0]); + strides.assign(1, strides[0]); + pads_begin.assign(1, pads_begin[0]); + pads_end.assign(1, pads_end[0]); } - #ifdef HAVE_OPENCL poolOp.release(); #endif