Merge pull request opencv#18234 from l-bat:onnx_reshape

alalek · alalek · commit 154380ccf5cf · 2020-09-02T19:31:22.000Z
diff --git 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;
     }
 

Original file line number	Diff line number	Diff line change
`@@ -335,6 +335,10 @@ void ONNXImporter::populateNet(Net dstNet)`
`335`	`335`	`{`
`336`	`336`	`inpShape[j] = tensorShape.dim(j).dim_value();`
`337`	`337`	`}`
	`338`	`+ if (!inpShape.empty())`
	`339`	`+ {`
	`340`	`+ inpShape[0] = std::max(inpShape[0], 1); // It's OK to have undetermined batch size`
	`341`	`+ }`
`338`	`342`	`outShapes[valueInfoProto.name()] = inpShape;`
`339`	`343`	`}`
`340`	`344`