Fix bug in ONNX Gather op

4 years ago · 6226ea0085
parent 1067cd0649
commit 6226ea0085
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);
                }