diff --git a/samples/dnn/tf_text_graph_ssd.py b/samples/dnn/tf_text_graph_ssd.py index 1576380646e5..35207ca3cdf4 100644 --- a/samples/dnn/tf_text_graph_ssd.py +++ b/samples/dnn/tf_text_graph_ssd.py @@ -35,8 +35,9 @@ def get(self, layer_id): 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