potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-29 07:13:02 +04:00
Code Packages Releases Wiki Activity

Merge pull request #17281 from l-bat:ssd_placeholder_shape

Browse Source
This commit is contained in:
Alexander Alekhin
2020-05-14 06:55:52 +00:00
parent b2464e3379 c5a2d28367
commit 6ced54a892
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
samples/dnn/tf_text_graph_ssd.py
+6
View File
@@ -234,6 +234,12 @@ def createSSDGraph(modelPath, configPath, outputPath):
# Connect input node to the first layer
assert(graph_def.node[0].op == 'Placeholder')
try:
input_shape = graph_def.node[0].attr['shape']['shape'][0]['dim']
input_shape[1]['size'] = image_height
input_shape[2]['size'] = image_width
except:
print("Input shapes are undefined")
# assert(graph_def.node[1].op == 'Conv2D')
weights = graph_def.node[1].input[-1]
for i in range(len(graph_def.node[1].input)):