diff --git a/modules/dnn/src/onnx/onnx_importer.cpp b/modules/dnn/src/onnx/onnx_importer.cpp index 8cb2c5eda1..153d026e4b 100644 --- a/modules/dnn/src/onnx/onnx_importer.cpp +++ b/modules/dnn/src/onnx/onnx_importer.cpp @@ -335,6 +335,10 @@ void ONNXImporter::populateNet(Net dstNet) { inpShape[j] = tensorShape.dim(j).dim_value(); } + if (!inpShape.empty()) + { + inpShape[0] = std::max(inpShape[0], 1); // It's OK to have undetermined batch size + } outShapes[valueInfoProto.name()] = inpShape; }