This commit is contained in:
Dmitry Kurtaev 2019-07-28 22:32:37 +03:00
parent 2693ed9b22
commit 59864a3bbc

View File

@ -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)