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

fix the infinite loop in tf importer.

Browse Source
This commit is contained in:
Zihao Mu 2022-10-31 20:10:25 +08:00
parent 87360c2ae5
commit 18fbb72f7d
1 changed files with 7 additions and 0 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

@ -829,12 +829,19 @@ void RemoveIdentityOps(tensorflow::GraphDef& net)
IdentityOpsMap::iterator it = identity_ops.find(input_op_name); IdentityOpsMap::iterator it = identity_ops.find(input_op_name);
if (it != identity_ops.end()) { if (it != identity_ops.end()) {
std::set<String> loopCheckSet;
// In case of Identity after Identity // In case of Identity after Identity
while (true) while (true)
{ {
IdentityOpsMap::iterator nextIt = identity_ops.find(it->second); IdentityOpsMap::iterator nextIt = identity_ops.find(it->second);
if (nextIt != identity_ops.end()) if (nextIt != identity_ops.end())
{
// Loop check
if (loopCheckSet.find(it->second) != loopCheckSet.end())
CV_Error(Error::StsError, "Found a loop in your input Tensorflow model, which is illegal!");
loopCheckSet.insert(it->second);
it = nextIt; it = nextIt;
}
else else
break; break;
} }