Merge pull request #18234 from l-bat:onnx_reshape

This commit is contained in:
Alexander Alekhin 2020-09-02 19:31:22 +00:00
commit 154380ccf5

View File

@ -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;
}