Skip to content

Commit

Permalink
bug fixed.
Browse files Browse the repository at this point in the history
  • Loading branch information
zihaomu committed Dec 10, 2022
1 parent 82efdb5 commit b905d6d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion modules/dnn/src/layers/convolution_layer.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1307,7 +1307,7 @@ class ConvolutionLayerImpl CV_FINAL : public BaseConvolutionLayerImpl
int C = inputs[0].size[1];

// Winograd only works when input h and w >= 12.
bool canUseWinograd = useWinograd && inputs[0].size[2] >= 12 && inputs[0].size[3] >= 12;
bool canUseWinograd = useWinograd && conv_dim == CONV_2D && inputs[0].size[2] >= 12 && inputs[0].size[3] >= 12;

CV_Assert(outputs[0].size[1] % ngroups == 0);
fastConvImpl = initFastConv(weightsMat, &biasvec[0], ngroups, K, C, kernel_size, strides,
Expand Down

0 comments on commit b905d6d

Please sign in to comment.