Merge pull request #18048 from l-bat:onnx_torchvision

336627a7 · Alexander Alekhin · 1067cd06 · 6226ea00 · 336627a7
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

modules/dnn/src/onnx/onnx_importer.cpp modules/dnn/src/onnx/onnx_importer.cpp +1 -0

未找到文件。
--- 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);
                }