mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-07-31 01:47:12 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #14501 from LaurentBerger:MatchLayer_text_graph_ssd

Browse Source
This commit is contained in:
Alexander Alekhin 2019-05-07 15:49:54 +00:00
commit b998c06d08
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
samples/dnn/tf_text_graph_ssd.py
View File

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