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

diff --git a/samples/dnn/tf_text_graph_ssd.py b/samples/dnn/tf_text_graph_ssd.py
index 1576380..35207ca 100644 (file)
--- 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