diff --git a/modules/dnn/src/op_cuda.hpp b/modules/dnn/src/op_cuda.hpp index 44e4955c50..880ed71634 100644 --- a/modules/dnn/src/op_cuda.hpp +++ b/modules/dnn/src/op_cuda.hpp @@ -305,6 +305,10 @@ namespace cv { namespace dnn { shape = shape_; offset = 0; shared_block = base->shared_block; + + auto numel = total(shape_); + if (numel > shared_block->device.size()) + shared_block->device.reset(numel); } static Ptr create(Mat& m) {