Fix priors boxes generation in case of non square inputs for TensorFlow SSDs

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