File tree Expand file tree Collapse file tree 1 file changed +7
-0
lines changed Expand file tree Collapse file tree 1 file changed +7
-0
lines changed Original file line number Diff line number Diff line change @@ -658,6 +658,8 @@ namespace cv {
658
658
if (pad)
659
659
padding = kernel_size / 2 ;
660
660
661
+ // Cannot divide 0
662
+ CV_Assert (stride > 0 );
661
663
CV_Assert (kernel_size > 0 && filters > 0 );
662
664
CV_Assert (tensor_shape[0 ] > 0 );
663
665
CV_Assert (tensor_shape[0 ] % groups == 0 );
@@ -690,6 +692,9 @@ namespace cv {
690
692
int kernel_size = getParam<int >(layer_params, " size" , 2 );
691
693
int stride = getParam<int >(layer_params, " stride" , 2 );
692
694
int padding = getParam<int >(layer_params, " padding" , kernel_size - 1 );
695
+ // Cannot divide 0
696
+ CV_Assert (stride > 0 );
697
+
693
698
setParams.setMaxpool (kernel_size, padding, stride);
694
699
695
700
tensor_shape[1 ] = (tensor_shape[1 ] - kernel_size + padding) / stride + 1 ;
@@ -732,6 +737,8 @@ namespace cv {
732
737
else if (layer_type == " reorg" )
733
738
{
734
739
int stride = getParam<int >(layer_params, " stride" , 2 );
740
+ // Cannot divide 0
741
+ CV_Assert (stride > 0 );
735
742
tensor_shape[0 ] = tensor_shape[0 ] * (stride * stride);
736
743
tensor_shape[1 ] = tensor_shape[1 ] / stride;
737
744
tensor_shape[2 ] = tensor_shape[2 ] / stride;
You can’t perform that action at this time.
0 commit comments