Skip to content

O
opencv
Commit f87a0dd6 authored by Dmitry Kurtaev's avatar Dmitry Kurtaev
Browse files
Options

Fix minimal aspect ratio scale for SSDs from TensorFlow

parent e9b033ef
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
samples/dnn/tf_text_graph_ssd.py
View file @ f87a0dd6
......@@ -196,8 +196,8 @@ for i in range(args.num_layers):
text_format.Merge('b: false', priorBox.attr["clip"])
if i == 0:
widths = [args.min_scale * 0.5, 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)]
widths = [0.1, 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:
widths = [scales[i] * sqrt(ar) for ar in args.aspect_ratios]
heights = [scales[i] / sqrt(ar) for ar in args.aspect_ratios]
......
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