potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-29 07:13:02 +04:00
Code Packages Releases Wiki Activity

add enableWinograd API for Net.

Browse Source
This commit is contained in:
Zihao Mu
2022-10-09 09:33:07 +08:00
parent 347246901e
commit 1e2ceca4df
15 changed files with 70 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/src/layers/convolution_layer.cpp
+3 -2
View File
@@ -89,7 +89,8 @@ public:
BaseConvolutionLayerImpl(const LayerParams &params)
{
setParamsFrom(params);
getConvolutionKernelParams(params, kernel_size, pads_begin, pads_end, strides, dilations, padMode, adjust_pads);
getConvolutionKernelParams(params, kernel_size, pads_begin, pads_end, strides, dilations,
padMode, adjust_pads, useWinograd);
numOutput = params.get<int>("num_output");
int ngroups = params.get<int>("group", 1);
@@ -2112,7 +2113,7 @@ public:
int dilation_w = dilations.back();
fastConv2dImpl = initFastConv2d(ngroups, K, C, Hk, Wk, stride_w, stride_h, dilation_w,
dilation_h, pads_begin, pads_end, weightsMat, &biasvec[0]);
dilation_h, pads_begin, pads_end, weightsMat, &biasvec[0], useWinograd);
}
if (fastConv2dImpl)