Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "cd00ba685b2162ef617146a938616dae7653872c"
Unverified Commit 79892d37 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Make ScaleJitter proportional (#5559)

parent 350a3e8e
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
references/detection/transforms.py
View file @ 79892d37
...@@ -310,9 +310,10 @@ class ScaleJitter(nn.Module): ...@@ -310,9 +310,10 @@ class ScaleJitter(nn.Module):
_, orig_height, orig_width = F.get_dimensions(image) _, orig_height, orig_width = F.get_dimensions(image)
r = self.scale_range[0] + torch.rand(1) * (self.scale_range[1] - self.scale_range[0]) scale = self.scale_range[0] + torch.rand(1) * (self.scale_range[1] - self.scale_range[0])
new_width = int(self.target_size[1] * r) r = min(self.target_size[1] / orig_height, self.target_size[0] / orig_width) * scale
new_height = int(self.target_size[0] * r) new_width = int(orig_width * r)
new_height = int(orig_height * r)
image = F.resize(image, [new_height, new_width], interpolation=self.interpolation) image = F.resize(image, [new_height, new_width], interpolation=self.interpolation)
......
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