mirror of
https://github.com/opencv/opencv.git
synced 2025-06-07 17:44:04 +08:00
Merge pull request #18234 from l-bat:onnx_reshape
This commit is contained in:
commit
154380ccf5
@ -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;
|
||||
}
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user