pull/15176/head
Dmitry Kurtaev 5 years ago
parent 2693ed9b22
commit 59864a3bbc
  1. 2
      samples/dnn/tf_text_graph_ssd.py

@ -67,7 +67,7 @@ def createSSDGraph(modelPath, configPath, outputPath):
'Sub', 'ResizeNearestNeighbor', 'Pad']
# Node with which prefixes should be removed
prefixesToRemove = ('MultipleGridAnchorGenerator/', 'Postprocessor/', 'Preprocessor/map')
prefixesToRemove = ('MultipleGridAnchorGenerator/', 'Concatenate/', 'Postprocessor/', 'Preprocessor/map')
# Load a config file.
config = readTextMessage(configPath)

Loading…
Cancel
Save