mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-18 08:05:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Determine SSD input shape

Browse Source
This commit is contained in:
Liubov Batanina 2020-05-13 14:44:23 +03:00
parent b2464e3379
commit c5a2d28367
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
samples/dnn/tf_text_graph_ssd.py
View File

@ -234,6 +234,12 @@ def createSSDGraph(modelPath, configPath, outputPath):
# Connect input node to the first layer # Connect input node to the first layer
assert(graph_def.node[0].op == 'Placeholder') 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') # assert(graph_def.node[1].op == 'Conv2D')
weights = graph_def.node[1].input[-1] weights = graph_def.node[1].input[-1]
for i in range(len(graph_def.node[1].input)): for i in range(len(graph_def.node[1].input)):