Unverified Commit f36f3514 authored by Nicolas Granger's avatar Nicolas Granger Committed by GitHub
Browse files

Fix #6550 (#6551)



Changes image size convention to (h, w). I don't think this is used by
models other than SSD which assumes this convention.
Co-authored-by: default avatarNicolas Granger <nicolas.granger@cea.fr>
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent 8acf1ca2
...@@ -213,8 +213,8 @@ class DefaultBoxGenerator(nn.Module): ...@@ -213,8 +213,8 @@ class DefaultBoxGenerator(nn.Module):
for k, f_k in enumerate(grid_sizes): for k, f_k in enumerate(grid_sizes):
# Now add the default boxes for each width-height pair # Now add the default boxes for each width-height pair
if self.steps is not None: if self.steps is not None:
x_f_k = image_size[0] / self.steps[k] x_f_k = image_size[1] / self.steps[k]
y_f_k = image_size[1] / self.steps[k] y_f_k = image_size[0] / self.steps[k]
else: else:
y_f_k, x_f_k = f_k y_f_k, x_f_k = f_k
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment