Commit 6e14cc21 authored by Dmitry Kurtaev's avatar Dmitry Kurtaev
parent 9e906d9e
......@@ -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')
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment