mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-10 02:53:07 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #19057 from sl-sergei:changes_for_cuda_1d

Browse Source
This commit is contained in:
Alexander Alekhin 2021-01-30 11:39:43 +00:00
commit b9dfffc237
2 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/dnn/src/layers/convolution_layer.cpp
View File

@ -117,6 +117,9 @@ public:
{ {
kernel_size.assign(1, kernel_size[0]); kernel_size.assign(1, kernel_size[0]);
strides.assign(1, strides[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); CV_Assert(weightShape.dims() == kernel_size.size() + 2);
for (int i = 0; i < kernel_size.size(); i++) { for (int i = 0; i < kernel_size.size(); i++) {

11
modules/dnn/src/layers/pooling_layer.cpp
View File

@ -167,14 +167,13 @@ public:
if (inputs[0].dims == 3) if (inputs[0].dims == 3)
{ {
//Pool1D // Pool1D
kernel_size.erase(kernel_size.begin() + 1); kernel_size.assign(1, kernel_size[0]);
strides.erase(strides.begin() + 1); strides.assign(1, strides[0]);
pads_begin.erase(pads_begin.begin() + 1); pads_begin.assign(1, pads_begin[0]);
pads_end.erase(pads_end.begin() + 1); pads_end.assign(1, pads_end[0]);
} }
#ifdef HAVE_OPENCL #ifdef HAVE_OPENCL
poolOp.release(); poolOp.release();
#endif #endif