Skip to content

O
opencv
Commit b998c06d authored by Alexander Alekhin's avatar Alexander Alekhin
Browse files
Options

Merge pull request #14501 from LaurentBerger:MatchLayer_text_graph_ssd

parents 5ba8aa78 fa8e301c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
samples/dnn/tf_text_graph_ssd.py
View file @ b998c06d
...@@ -274,7 +274,8 @@ def createSSDGraph(modelPath, configPath, outputPath): ...@@ -274,7 +274,8 @@ def createSSDGraph(modelPath, configPath, outputPath):
num_matched_layers = 0 num_matched_layers = 0
for node in graph_def.node: 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): re.match('WeightSharedConvolutionalBoxPredictor(_\d)*/BoxPredictor/Conv2D', node.name):
node.addAttr('loc_pred_transposed', True) node.addAttr('loc_pred_transposed', True)
num_matched_layers += 1 num_matched_layers += 1
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment