Make ScaleJitter proportional (#5559)

79892d37 · Vasilis Vryniotis · GitHub · 350a3e8e · 79892d37
Unverified Commit 79892d37 authored Mar 07, 2022 by Vasilis Vryniotis Committed by GitHub Mar 07, 2022
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

references/detection/transforms.py references/detection/transforms.py +4 -3

No files found.
--- a/references/detection/transforms.py
+++ b/references/detection/transforms.py
@@ -310,9 +310,10 @@ class ScaleJitter(nn.Module):

        _, orig_height, orig_width = F.get_dimensions(image)

-        r = self.scale_range[0] + torch.rand(1) * (self.scale_range[1] - self.scale_range[0])
-        new_width = int(self.target_size[1] * r)
-        new_height = int(self.target_size[0] * r)
+        scale = self.scale_range[0] + torch.rand(1) * (self.scale_range[1] - self.scale_range[0])
+        r = min(self.target_size[1] / orig_height, self.target_size[0] / orig_width) * scale
+        new_width = int(orig_width * r)
+        new_height = int(orig_height * r)

        image = F.resize(image, [new_height, new_width], interpolation=self.interpolation)