This commit is contained in:
Dmitry Kurtaev
2019-07-28 22:32:37 +03:00
parent 2693ed9b22
commit 59864a3bbc
+1 -1
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)