Merge pull request #11180 from dkurt:fix_dnn_tf_ssd

pull/11187/head
Alexander Alekhin 7 years ago
commit c37d9573fc
  1. 4
      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]

Loading…
Cancel
Save