mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-08-05 22:19:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixed removePhaseSwitches in tf_graph_simplifier

Browse Source
This commit is contained in:
Alexander Lyulkov 2023-09-15 14:22:21 +07:00
parent 6694d87a23
commit d4cb564ce2
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
modules/dnn/src/tensorflow/tf_graph_simplifier.cpp
View File

@ -1120,15 +1120,16 @@ void removePhaseSwitches(tensorflow::GraphDef& net)
inpName = inpName.substr(1 + (int)inpName.find('^'), inpName.rfind(':'));
nodesMapIt = nodesMap.find(inpName);
CV_Assert(nodesMapIt != nodesMap.end());
int inpNodeId = nodesMapIt->second;
CV_CheckGT(numConsumers[inpNodeId], 0,
"Input node of the current node should have at least one output node");
if (numConsumers[inpNodeId] == 1)
{
mergeOpSubgraphNodes.push(inpNodeId);
nodesToRemove.push_back(inpNodeId);
}
else if (numConsumers[inpNodeId] > 0)
numConsumers[inpNodeId] -= 1;
numConsumers[inpNodeId] -= 1;
}
}
std::sort(nodesToRemove.begin(), nodesToRemove.end());