Support Reshape with zero dim

pull/18234/head
Liubov Batanina 5 years ago
parent 7495a4722f
commit 2349a09736
  1. 4
      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;
}

Loading…
Cancel
Save