Merge pull request #18048 from l-bat:onnx_torchvision

5 years ago · 336627a776
parent 1067cd0649 6226ea0085
commit 336627a776
1 changed files with 1 additions and 0 deletions
--- a/modules/dnn/src/onnx/onnx_importer.cpp
+++ b/modules/dnn/src/onnx/onnx_importer.cpp
@ -1395,6 +1395,7 @@ void ONNXImporter::populateNet(Net dstNet)

                    inpShape.erase(inpShape.begin() + axis);
                    layerParams.type = "Reshape";
+                    layerParams.set("axis", 0);
                    layerParams.set("dim", DictValue::arrayInt(&inpShape[0], inpShape.size()));
                    node_proto.set_input(0, sliceLp.name);
                }