Commit fe973cee authored by Ryan Peach's avatar Ryan Peach Committed by Francisco Massa
Browse files

Removed the +1 and divide by 2 to allow for random offset padding (#564)

parent be68e244
...@@ -445,10 +445,10 @@ class RandomCrop(object): ...@@ -445,10 +445,10 @@ class RandomCrop(object):
# pad the width if needed # pad the width if needed
if self.pad_if_needed and img.size[0] < self.size[1]: if self.pad_if_needed and img.size[0] < self.size[1]:
img = F.pad(img, (int((1 + self.size[1] - img.size[0]) / 2), 0), self.fill, self.padding_mode) img = F.pad(img, (self.size[1] - img.size[0], 0), self.fill, self.padding_mode)
# pad the height if needed # pad the height if needed
if self.pad_if_needed and img.size[1] < self.size[0]: if self.pad_if_needed and img.size[1] < self.size[0]:
img = F.pad(img, (0, int((1 + self.size[0] - img.size[1]) / 2)), self.fill, self.padding_mode) img = F.pad(img, (0, self.size[0] - img.size[1]), self.fill, self.padding_mode)
i, j, h, w = self.get_params(img, self.size) i, j, h, w = self.get_params(img, self.size)
......
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