From fa8e301c6984e614f0e088aa06a0a75d34abecf9 Mon Sep 17 00:00:00 2001 From: LaurentBerger Date: Tue, 7 May 2019 16:11:08 +0200 Subject: [PATCH] extends regex matching for conv group of layers --- samples/dnn/tf_text_graph_ssd.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/samples/dnn/tf_text_graph_ssd.py b/samples/dnn/tf_text_graph_ssd.py index 35207ca..730e0bb 100644 --- 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 -- 2.7.4