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

Fix minimal aspect ratio scale for SSDs from TensorFlow

Browse Source
This commit is contained in:
Dmitry Kurtaev 2018-03-28 12:53:12 +03:00
parent e9b033ef3a
commit f87a0dd6a7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
samples/dnn/tf_text_graph_ssd.py
View File

@ -196,8 +196,8 @@ for i in range(args.num_layers):
text_format.Merge('b: false', priorBox.attr["clip"]) text_format.Merge('b: false', priorBox.attr["clip"])
if i == 0: if i == 0:
widths = [args.min_scale * 0.5, args.min_scale * sqrt(2.0), args.min_scale * sqrt(0.5)] widths = [0.1, args.min_scale * sqrt(2.0), args.min_scale * sqrt(0.5)]
heights = [args.min_scale * 0.5, args.min_scale / sqrt(2.0), args.min_scale / sqrt(0.5)] heights = [0.1, args.min_scale / sqrt(2.0), args.min_scale / sqrt(0.5)]
else: else:
widths = [scales[i] * sqrt(ar) for ar in args.aspect_ratios] widths = [scales[i] * sqrt(ar) for ar in args.aspect_ratios]
heights = [scales[i] / sqrt(ar) for ar in args.aspect_ratios] heights = [scales[i] / sqrt(ar) for ar in args.aspect_ratios]