Merge pull request #23671 from zihaomu:fix_potential_bug

DNN: fix potential bug, stride should not be set as 0.
2 years ago · 26a7b332cb
parent f07b01cc34 4384e77bd1
commit 26a7b332cb
1 changed files with 2 additions and 2 deletions
--- a/modules/dnn/src/layers/cpu_kernels/convolution.cpp
+++ b/modules/dnn/src/layers/cpu_kernels/convolution.cpp
@ -54,8 +54,8 @@ Ptr<FastConv> initFastConv(
    conv->pad_bottom = conv_dim == CONV_1D ? 0 : (int)pads_end[pads_end.size() - 2];
    conv->pad_right = (int)pads_end.back();

-    int stride_d = conv_dim == CONV_3D ? (int)strides[0] : 0;
-    int stride_h = conv_dim == CONV_1D ? 0 : (int)strides[strides.size() - 2];
+    int stride_d = conv_dim == CONV_3D ? (int)strides[0] : 1;
+    int stride_h = conv_dim == CONV_1D ? 1 : (int)strides[strides.size() - 2];
    int stride_w = (int)strides.back();

    int dilation_d = conv_dim == CONV_3D ? (int)dilations[0] : 1;