Skip to content

Commit 9aff567

Browse files
authored
Revert "add padding-mode choice to RandomCrop (#512)" (#515)
This reverts commit 11da4f1.
1 parent 11da4f1 commit 9aff567

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

torchvision/transforms/transforms.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -375,19 +375,16 @@ class RandomCrop(object):
375375
of the image. Default is 0, i.e no padding. If a sequence of length
376376
4 is provided, it is used to pad left, top, right, bottom borders
377377
respectively.
378-
padding_mode (str): Type of padding. Should be: constant, edge, reflect
379-
or symmetric. Default is constant.
380378
pad_if_needed (boolean): It will pad the image if smaller than the
381379
desired size to avoid raising an exception.
382380
"""
383381

384-
def __init__(self, size, padding=0, padding_mode='constant', pad_if_needed=False):
382+
def __init__(self, size, padding=0, pad_if_needed=False):
385383
if isinstance(size, numbers.Number):
386384
self.size = (int(size), int(size))
387385
else:
388386
self.size = size
389387
self.padding = padding
390-
self.padding_mode = padding_mode
391388
self.pad_if_needed = pad_if_needed
392389

393390
@staticmethod
@@ -419,14 +416,14 @@ def __call__(self, img):
419416
PIL Image: Cropped image.
420417
"""
421418
if self.padding > 0:
422-
img = F.pad(img, self.padding, padding_mode=self.padding_mode)
419+
img = F.pad(img, self.padding)
423420

424421
# pad the width if needed
425422
if self.pad_if_needed and img.size[0] < self.size[1]:
426-
img = F.pad(img, (int((1 + self.size[1] - img.size[0]) / 2), 0), padding_mode=self.padding_mode)
423+
img = F.pad(img, (int((1 + self.size[1] - img.size[0]) / 2), 0))
427424
# pad the height if needed
428425
if self.pad_if_needed and img.size[1] < self.size[0]:
429-
img = F.pad(img, (0, int((1 + self.size[0] - img.size[1]) / 2)), padding_mode=self.padding_mode)
426+
img = F.pad(img, (0, int((1 + self.size[0] - img.size[1]) / 2)))
430427

431428
i, j, h, w = self.get_params(img, self.size)
432429

0 commit comments

Comments
 (0)