extends regex matching for conv group of layers

6 years ago · fa8e301c69
parent 94c7d99d0d
commit fa8e301c69
1 changed files with 2 additions and 1 deletions
--- a/samples/dnn/tf_text_graph_ssd.py
+++ b/samples/dnn/tf_text_graph_ssd.py
@ -274,7 +274,8 @@ def createSSDGraph(modelPath, configPath, outputPath):

    num_matched_layers = 0
    for node in graph_def.node:
-        if re.match('BoxPredictor_\d/BoxEncodingPredictor/Conv2D', node.name) or \
+        if re.match('BoxPredictor_\d/BoxEncodingPredictor/convolution', node.name) or \
+           re.match('BoxPredictor_\d/BoxEncodingPredictor/Conv2D', node.name) or \
           re.match('WeightSharedConvolutionalBoxPredictor(_\d)*/BoxPredictor/Conv2D', node.name):
            node.addAttr('loc_pred_transposed', True)
            num_matched_layers += 1