Fix minimal aspect ratio scale for SSDs from TensorFlow

7 years ago · f87a0dd6a7
parent e9b033ef3a
commit f87a0dd6a7
1 changed files with 2 additions and 2 deletions
--- a/samples/dnn/tf_text_graph_ssd.py
+++ b/samples/dnn/tf_text_graph_ssd.py
@ -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]