dnn: fix sporadic crashes in getUMat()

Incorrect "total" buffer size calculated in StdMatAllocator::allocate() due wrong step values.
pull/9939/head
Alexander Alekhin 7 years ago
parent 21c8e6d02d
commit 436a1f72a5
  1. 3
      modules/dnn/src/tensorflow/tf_importer.cpp

@ -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]);

Loading…
Cancel
Save