Commit 436a1f72 authored by Alexander Alekhin's avatar Alexander Alekhin

dnn: fix sporadic crashes in getUMat()

Incorrect "total" buffer size calculated in StdMatAllocator::allocate() due wrong step values.
parent 21c8e6d0
......@@ -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]);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment