Merge pull request #17419 from YourButterfly:fix_divide_zero_in_darknet_3_4

modules/dnn/src/darknet/darknet_io.cpp

@@ -658,6 +658,8 @@ namespace cv {
                         if (pad)
                             padding = kernel_size / 2;
 
+                        // Cannot divide 0
+                        CV_Assert(stride > 0);
                         CV_Assert(kernel_size > 0 && filters > 0);
                         CV_Assert(tensor_shape[0] > 0);
                         CV_Assert(tensor_shape[0] % groups == 0);
@@ -690,6 +692,9 @@ namespace cv {
                         int kernel_size = getParam<int>(layer_params, "size", 2);
                         int stride = getParam<int>(layer_params, "stride", 2);
                         int padding = getParam<int>(layer_params, "padding", kernel_size - 1);
+                        // Cannot divide 0
+                        CV_Assert(stride > 0);
+
                         setParams.setMaxpool(kernel_size, padding, stride);
 
                         tensor_shape[1] = (tensor_shape[1] - kernel_size + padding) / stride + 1;
@@ -732,6 +737,8 @@ namespace cv {
                     else if (layer_type == "reorg")
                     {
                         int stride = getParam<int>(layer_params, "stride", 2);
+                        // Cannot divide 0
+                        CV_Assert(stride > 0);
                         tensor_shape[0] = tensor_shape[0] * (stride * stride);
                         tensor_shape[1] = tensor_shape[1] / stride;
                         tensor_shape[2] = tensor_shape[2] / stride;