mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-11-29 05:29:54 +08:00
Code Issues Actions 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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
samples/dnn/tf_text_graph_ssd.py
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