Skip to content

Commit 7608ee2

Browse files
committed
Update arguments.
1 parent 44ac8fd commit 7608ee2

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

torchvision/transforms/transforms.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1058,7 +1058,11 @@ def get_params(brightness: Optional[List[float]],
10581058
) -> Tuple[Tensor, Optional[float], Optional[float], Optional[float], Optional[float]]:
10591059
"""Get the parameters for the randomized transform to be applied on image.
10601060
1061-
Arguments are same as that of __init__.
1061+
Args:
1062+
brightness (tuple of float (min, max)): The range from which the brightness_factor is chosen uniformly.
1063+
contrast (tuple of float (min, max)): The range from which the contrast_factor is chosen uniformly.
1064+
saturation (tuple of float (min, max)): The range from which the saturation_factor is chosen uniformly.
1065+
hue (tuple of float (min, max)): The range from which the hue_factor is chosen uniformly.
10621066
10631067
Returns:
10641068
tuple: The parameters used to apply the randomized transform

0 commit comments

Comments
 (0)