@@ -375,19 +375,16 @@ class RandomCrop(object):
375
375
of the image. Default is 0, i.e no padding. If a sequence of length
376
376
4 is provided, it is used to pad left, top, right, bottom borders
377
377
respectively.
378
- padding_mode (str): Type of padding. Should be: constant, edge, reflect
379
- or symmetric. Default is constant.
380
378
pad_if_needed (boolean): It will pad the image if smaller than the
381
379
desired size to avoid raising an exception.
382
380
"""
383
381
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 ):
385
383
if isinstance (size , numbers .Number ):
386
384
self .size = (int (size ), int (size ))
387
385
else :
388
386
self .size = size
389
387
self .padding = padding
390
- self .padding_mode = padding_mode
391
388
self .pad_if_needed = pad_if_needed
392
389
393
390
@staticmethod
@@ -419,14 +416,14 @@ def __call__(self, img):
419
416
PIL Image: Cropped image.
420
417
"""
421
418
if self .padding > 0 :
422
- img = F .pad (img , self .padding , padding_mode = self . padding_mode )
419
+ img = F .pad (img , self .padding )
423
420
424
421
# pad the width if needed
425
422
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 ))
427
424
# pad the height if needed
428
425
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 )))
430
427
431
428
i , j , h , w = self .get_params (img , self .size )
432
429
0 commit comments