1
0
mirror of https://github.com/opencv/opencv.git synced 2026-07-29 07:13:02 +04:00

Merge pull request #13906 from dkurt:fix_tf_ssd_min_size

This commit is contained in:
Alexander Alekhin
2019-02-25 16:09:57 +00:00
+3 -2
View File
@@ -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