mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-11-27 20:50:25 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix bug in ONNX Gather op

Browse Source
This commit is contained in:
Liubov Batanina 2020-08-06 15:47:34 +03:00
parent 1067cd0649
commit 6226ea0085
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
modules/dnn/src/onnx/onnx_importer.cpp
View File

@ -1395,6 +1395,7 @@ void ONNXImporter::populateNet(Net dstNet)
inpShape.erase(inpShape.begin() + axis); inpShape.erase(inpShape.begin() + axis);
layerParams.type = "Reshape"; layerParams.type = "Reshape";
layerParams.set("axis", 0);
layerParams.set("dim", DictValue::arrayInt(&inpShape[0], inpShape.size())); layerParams.set("dim", DictValue::arrayInt(&inpShape[0], inpShape.size()));
node_proto.set_input(0, sliceLp.name); node_proto.set_input(0, sliceLp.name);
} }