diff --git a/modules/dnn/src/tensorflow/tf_importer.cpp b/modules/dnn/src/tensorflow/tf_importer.cpp index 78a42ca1c7..e266d3b464 100644 --- a/modules/dnn/src/tensorflow/tf_importer.cpp +++ b/modules/dnn/src/tensorflow/tf_importer.cpp @@ -1585,7 +1585,7 @@ void TFImporter::populateNet(Net dstNet) } } } - else if (type == "FusedBatchNorm") + else if (type == "FusedBatchNorm" || type == "FusedBatchNormV3") { // op: "FusedBatchNorm" // input: "input" diff --git a/samples/dnn/tf_text_graph_ssd.py b/samples/dnn/tf_text_graph_ssd.py index 905f751557..1060047260 100644 --- a/samples/dnn/tf_text_graph_ssd.py +++ b/samples/dnn/tf_text_graph_ssd.py @@ -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')