potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-29 15:23:05 +04:00
Code Packages Releases Wiki Activity

Merge pull request #24275 from alexlyulkov:al/fix-tf-graph-simplifier

Browse Source 
Fixed removePhaseSwitches in tf_graph_simplifier
This commit is contained in:
Alexander Smorkalov
2023-09-18 11:02:44 +03:00
committed by GitHub
parent 0a53afe8ba d4cb564ce2
commit 157b0e7760
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/src/tensorflow/tf_graph_simplifier.cpp
+4 -3
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());