diff --git a/modules/dnn/src/tensorflow/tf_importer.cpp b/modules/dnn/src/tensorflow/tf_importer.cpp index 8e1f18e5b3..e640938100 100644 --- a/modules/dnn/src/tensorflow/tf_importer.cpp +++ b/modules/dnn/src/tensorflow/tf_importer.cpp @@ -363,7 +363,7 @@ void RemoveIdentityOps(tensorflow::GraphDef& net) { const tensorflow::NodeDef &layer = net.node(li); String type = layer.op(); - if (type == "Identity") { + if (type == "Identity" || type == "Dropout") { identity_ops_idx.push_back(li); identity_ops[layer.name()] = layer.input(0); } diff --git a/modules/dnn/test/test_tf_importer.cpp b/modules/dnn/test/test_tf_importer.cpp index f296fc4289..948262769e 100644 --- a/modules/dnn/test/test_tf_importer.cpp +++ b/modules/dnn/test/test_tf_importer.cpp @@ -138,6 +138,11 @@ TEST(Test_TensorFlow, matmul) runTensorFlowNet("matmul"); } +TEST(Test_TensorFlow, defun) +{ + runTensorFlowNet("defun_dropout"); +} + TEST(Test_TensorFlow, fp16) { const float l1 = 1e-3;