Unverified Commit 39772ece authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

remove option to pass fill as str (#5632)


Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent 585b3b13
...@@ -448,11 +448,11 @@ def pad(img: Tensor, padding: List[int], fill: int = 0, padding_mode: str = "con ...@@ -448,11 +448,11 @@ def pad(img: Tensor, padding: List[int], fill: int = 0, padding_mode: str = "con
.. note:: .. note::
In torchscript mode padding as single int is not supported, use a sequence of In torchscript mode padding as single int is not supported, use a sequence of
length 1: ``[padding, ]``. length 1: ``[padding, ]``.
fill (number or str or tuple): Pixel fill value for constant fill. Default is 0. fill (number or tuple): Pixel fill value for constant fill. Default is 0.
If a tuple of length 3, it is used to fill R, G, B channels respectively. If a tuple of length 3, it is used to fill R, G, B channels respectively.
This value is only used when the padding_mode is constant. This value is only used when the padding_mode is constant.
Only number is supported for torch Tensor. Only number is supported for torch Tensor.
Only int or str or tuple value is supported for PIL Image. Only int or tuple value is supported for PIL Image.
padding_mode (str): Type of padding. Should be: constant, edge, reflect or symmetric. padding_mode (str): Type of padding. Should be: constant, edge, reflect or symmetric.
Default is constant. Default is constant.
......
...@@ -154,7 +154,7 @@ def pad( ...@@ -154,7 +154,7 @@ def pad(
if not isinstance(padding, (numbers.Number, tuple, list)): if not isinstance(padding, (numbers.Number, tuple, list)):
raise TypeError("Got inappropriate padding arg") raise TypeError("Got inappropriate padding arg")
if not isinstance(fill, (numbers.Number, str, tuple, list)): if not isinstance(fill, (numbers.Number, tuple, list)):
raise TypeError("Got inappropriate fill arg") raise TypeError("Got inappropriate fill arg")
if not isinstance(padding_mode, str): if not isinstance(padding_mode, str):
raise TypeError("Got inappropriate padding_mode arg") raise TypeError("Got inappropriate padding_mode arg")
......
...@@ -400,11 +400,11 @@ class Pad(torch.nn.Module): ...@@ -400,11 +400,11 @@ class Pad(torch.nn.Module):
.. note:: .. note::
In torchscript mode padding as single int is not supported, use a sequence of In torchscript mode padding as single int is not supported, use a sequence of
length 1: ``[padding, ]``. length 1: ``[padding, ]``.
fill (number or str or tuple): Pixel fill value for constant fill. Default is 0. If a tuple of fill (number or tuple): Pixel fill value for constant fill. Default is 0. If a tuple of
length 3, it is used to fill R, G, B channels respectively. length 3, it is used to fill R, G, B channels respectively.
This value is only used when the padding_mode is constant. This value is only used when the padding_mode is constant.
Only number is supported for torch Tensor. Only number is supported for torch Tensor.
Only int or str or tuple value is supported for PIL Image. Only int or tuple value is supported for PIL Image.
padding_mode (str): Type of padding. Should be: constant, edge, reflect or symmetric. padding_mode (str): Type of padding. Should be: constant, edge, reflect or symmetric.
Default is constant. Default is constant.
...@@ -428,7 +428,7 @@ class Pad(torch.nn.Module): ...@@ -428,7 +428,7 @@ class Pad(torch.nn.Module):
if not isinstance(padding, (numbers.Number, tuple, list)): if not isinstance(padding, (numbers.Number, tuple, list)):
raise TypeError("Got inappropriate padding arg") raise TypeError("Got inappropriate padding arg")
if not isinstance(fill, (numbers.Number, str, tuple, list)): if not isinstance(fill, (numbers.Number, tuple, list)):
raise TypeError("Got inappropriate fill arg") raise TypeError("Got inappropriate fill arg")
if padding_mode not in ["constant", "edge", "reflect", "symmetric"]: if padding_mode not in ["constant", "edge", "reflect", "symmetric"]:
...@@ -595,11 +595,11 @@ class RandomCrop(torch.nn.Module): ...@@ -595,11 +595,11 @@ class RandomCrop(torch.nn.Module):
pad_if_needed (boolean): It will pad the image if smaller than the pad_if_needed (boolean): It will pad the image if smaller than the
desired size to avoid raising an exception. Since cropping is done desired size to avoid raising an exception. Since cropping is done
after padding, the padding seems to be done at a random offset. after padding, the padding seems to be done at a random offset.
fill (number or str or tuple): Pixel fill value for constant fill. Default is 0. If a tuple of fill (number or tuple): Pixel fill value for constant fill. Default is 0. If a tuple of
length 3, it is used to fill R, G, B channels respectively. length 3, it is used to fill R, G, B channels respectively.
This value is only used when the padding_mode is constant. This value is only used when the padding_mode is constant.
Only number is supported for torch Tensor. Only number is supported for torch Tensor.
Only int or str or tuple value is supported for PIL Image. Only int or tuple value is supported for PIL Image.
padding_mode (str): Type of padding. Should be: constant, edge, reflect or symmetric. padding_mode (str): Type of padding. Should be: constant, edge, reflect or symmetric.
Default is constant. Default is constant.
......
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