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

Import wrapped Dropout subgraphs from TensorFlow

Browse Source
This commit is contained in:
Dmitry Kurtaev
2017-09-20 13:30:25 +03:00
parent 6a5298a532
commit bdd8cc697a
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/src/tensorflow/tf_importer.cpp
+1 -1
View File
@@ -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);
}