Merge pull request #15986 from dkurt:fix_15863

pull/15994/head
Alexander Alekhin 5 years ago
commit a093a0e05c
  1. 2
      modules/dnn/src/tensorflow/tf_importer.cpp
  2. 2
      samples/dnn/tf_text_graph_ssd.py

@ -1585,7 +1585,7 @@ void TFImporter::populateNet(Net dstNet)
}
}
}
else if (type == "FusedBatchNorm")
else if (type == "FusedBatchNorm" || type == "FusedBatchNormV3")
{
// op: "FusedBatchNorm"
// input: "input"

@ -64,7 +64,7 @@ def createSSDGraph(modelPath, configPath, outputPath):
# Nodes that should be kept.
keepOps = ['Conv2D', 'BiasAdd', 'Add', 'Relu', 'Relu6', 'Placeholder', 'FusedBatchNorm',
'DepthwiseConv2dNative', 'ConcatV2', 'Mul', 'MaxPool', 'AvgPool', 'Identity',
'Sub', 'ResizeNearestNeighbor', 'Pad']
'Sub', 'ResizeNearestNeighbor', 'Pad', 'FusedBatchNormV3']
# Node with which prefixes should be removed
prefixesToRemove = ('MultipleGridAnchorGenerator/', 'Concatenate/', 'Postprocessor/', 'Preprocessor/map')

Loading…
Cancel
Save