Update transforms.py (#31252)

update RandomHorizontal Flip random description and examples
test_model_benchmark_ci
Bin Lu 4 years ago committed by GitHub
parent 5610c1717e
commit a37658daff
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -537,7 +537,7 @@ class RandomHorizontalFlip(BaseTransform):
"""Horizontally flip the input data randomly with a given probability. """Horizontally flip the input data randomly with a given probability.
Args: Args:
prob (float, optional): Probability of the input data being flipped. Default: 0.5 prob (float, optional): Probability of the input data being flipped. Should be in [0, 1]. Default: 0.5
keys (list[str]|tuple[str], optional): Same as ``BaseTransform``. Default: None. keys (list[str]|tuple[str], optional): Same as ``BaseTransform``. Default: None.
Examples: Examples:
@ -548,7 +548,7 @@ class RandomHorizontalFlip(BaseTransform):
from PIL import Image from PIL import Image
from paddle.vision.transforms import RandomHorizontalFlip from paddle.vision.transforms import RandomHorizontalFlip
transform = RandomHorizontalFlip(224) transform = RandomHorizontalFlip(0.5)
fake_img = Image.fromarray((np.random.rand(300, 320, 3) * 255.).astype(np.uint8)) fake_img = Image.fromarray((np.random.rand(300, 320, 3) * 255.).astype(np.uint8))

Loading…
Cancel
Save