Skip to content

Commit fe973ce

Browse files
ryanpeachfmassa
authored andcommitted
Removed the +1 and divide by 2 to allow for random offset padding (#564)
1 parent be68e24 commit fe973ce

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchvision/transforms/transforms.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -445,10 +445,10 @@ def __call__(self, img):
445445

446446
# pad the width if needed
447447
if self.pad_if_needed and img.size[0] < self.size[1]:
448-
img = F.pad(img, (int((1 + self.size[1] - img.size[0]) / 2), 0), self.fill, self.padding_mode)
448+
img = F.pad(img, (self.size[1] - img.size[0], 0), self.fill, self.padding_mode)
449449
# pad the height if needed
450450
if self.pad_if_needed and img.size[1] < self.size[0]:
451-
img = F.pad(img, (0, int((1 + self.size[0] - img.size[1]) / 2)), self.fill, self.padding_mode)
451+
img = F.pad(img, (0, self.size[0] - img.size[1]), self.fill, self.padding_mode)
452452

453453
i, j, h, w = self.get_params(img, self.size)
454454

0 commit comments

Comments
 (0)