Skip to content

Commit 6eebdef

Browse files
committed
fix a bug described in issue #488
1 parent f87a896 commit 6eebdef

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
@@ -372,7 +372,7 @@ class RandomCrop(object):
372372
desired size to avoid raising an exception.
373373
"""
374374

375-
def __init__(self, size, padding=0, pad_if_needed=False):
375+
def __init__(self, size, padding=None, pad_if_needed=False):
376376
if isinstance(size, numbers.Number):
377377
self.size = (int(size), int(size))
378378
else:
@@ -408,7 +408,7 @@ def __call__(self, img):
408408
Returns:
409409
PIL Image: Cropped image.
410410
"""
411-
if self.padding > 0:
411+
if self.padding is not None:
412412
img = F.pad(img, self.padding)
413413

414414
# pad the width if needed

0 commit comments

Comments
 (0)