mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-08-06 14:36:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #9939 from alalek:fix_dnn_getUMat_crash

Browse Source
This commit is contained in:
Vadim Pisarevsky 2017-10-27 11:06:22 +00:00
commit 7b8fb64f21
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/dnn/src/tensorflow/tf_importer.cpp
View File

@ -692,8 +692,11 @@ void TFImporter::populateNet(Net dstNet)
int dst_i = (j * chMultiplier + i) * height* width + s;
dst[dst_i] = src[src_i];
}
// TODO Use reshape instead
kshape[0] = inCh * chMultiplier;
kshape[1] = 1;
size_t* kstep = layerParams.blobs[0].step.p;
kstep[0] = kstep[1]; // fix steps too
}
layerParams.set("kernel_h", kshape[2]);
layerParams.set("kernel_w", kshape[3]);