mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-08-06 14:36:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Resolve https://github.com/opencv/opencv/issues/15863

Browse Source
This commit is contained in:
Dmitry Kurtaev 2019-11-24 21:59:25 +03:00
parent 9e906d9e21
commit 6e14cc2189
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/dnn/src/tensorflow/tf_importer.cpp
View File

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

2
samples/dnn/tf_text_graph_ssd.py
View File

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