third_party/opencv
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Dmitry Kurtaev
2019-02-25 12:43:42 +03:00
parent 1db5d82b7f
commit fe7c93cc05
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
samples/dnn/tf_text_graph_ssd.py
+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