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

Merge pull request #21163 from rogday:transpose_default

Browse Source
This commit is contained in:
Alexander Alekhin 2021-12-01 22:47:57 +00:00
commit a806e8cc58
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1645,6 +1645,16 @@ void ONNXImporter::parseTranspose(LayerParams& layerParams, const opencv_onnx::N
{ {
layerParams.type = "Permute"; layerParams.type = "Permute";
replaceLayerParam(layerParams, "perm", "order"); replaceLayerParam(layerParams, "perm", "order");
if (!layerParams.has("order")) {
MatShape inpShape = outShapes[node_proto.input(0)];
size_t dims = inpShape.size();
std::vector<int> perm(dims);
for (size_t d = 0; d < dims; ++d)
{
perm[d] = static_cast<int>(dims - 1 - d);
}
layerParams.set("order", DictValue::arrayInt(perm.data(), perm.size()));
}
CV_Assert(node_proto.input_size() == 1); CV_Assert(node_proto.input_size() == 1);
if (constBlobs.find(node_proto.input(0)) != constBlobs.end()) if (constBlobs.find(node_proto.input(0)) != constBlobs.end())