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

Merge pull request #18048 from l-bat:onnx_torchvision

Browse Source
This commit is contained in:
Alexander Alekhin 2020-08-06 20:21:47 +00:00
commit 336627a776
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);
} }