Merge pull request opencv#18048 from l-bat:onnx_torchvision

alalek · alalek · commit 336627a7765d · 2020-08-06T20:21:47.000Z
diff --git 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);
                 }

Original file line number	Diff line number	Diff line change
`@@ -1395,6 +1395,7 @@ void ONNXImporter::populateNet(Net dstNet)`
`1395`	`1395`
`1396`	`1396`	`inpShape.erase(inpShape.begin() + axis);`
`1397`	`1397`	`layerParams.type = "Reshape";`
	`1398`	`+ layerParams.set("axis", 0);`
`1398`	`1399`	`layerParams.set("dim", DictValue::arrayInt(&inpShape[0], inpShape.size()));`
`1399`	`1400`	`node_proto.set_input(0, sliceLp.name);`
`1400`	`1401`	`}`