Commit 0aa0f2ad authored by Alexander Alekhin's avatar Alexander Alekhin

Merge pull request #13906 from dkurt:fix_tf_ssd_min_size

parents f6d3adcf fe7c93cc
......@@ -35,8 +35,9 @@ class SSDAnchorGenerator:
widths += [sqrt(self.scales[layer_id] * self.scales[layer_id + 1])]
heights += [sqrt(self.scales[layer_id] * self.scales[layer_id + 1])]
widths = [w * self.image_width for w in widths]
heights = [h * self.image_height for h in heights]
min_size = min(self.image_width, self.image_height)
widths = [w * min_size for w in widths]
heights = [h * min_size for h in heights]
return widths, heights
......
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