From 49ea181f64bb362d9ce30df2cf19585688746dbb Mon Sep 17 00:00:00 2001 From: Georg Grab Date: Mon, 14 Mar 2022 15:04:27 +0100 Subject: [PATCH] fix typo in docstrings of some transforms --- torchvision/transforms/transforms.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torchvision/transforms/transforms.py b/torchvision/transforms/transforms.py index a95977ad704..a5081390f2d 100644 --- a/torchvision/transforms/transforms.py +++ b/torchvision/transforms/transforms.py @@ -1894,7 +1894,7 @@ class RandomPosterize(torch.nn.Module): Args: bits (int): number of bits to keep for each channel (0-8) - p (float): probability of the image being color inverted. Default value is 0.5 + p (float): probability of the image being posterized. Default value is 0.5 """ def __init__(self, bits, p=0.5): @@ -1927,7 +1927,7 @@ class RandomSolarize(torch.nn.Module): Args: threshold (float): all pixels equal or above this value are inverted. - p (float): probability of the image being color inverted. Default value is 0.5 + p (float): probability of the image being solarized. Default value is 0.5 """ def __init__(self, threshold, p=0.5): @@ -1960,7 +1960,7 @@ class RandomAdjustSharpness(torch.nn.Module): sharpness_factor (float): How much to adjust the sharpness. Can be any non negative number. 0 gives a blurred image, 1 gives the original image while 2 increases the sharpness by a factor of 2. - p (float): probability of the image being color inverted. Default value is 0.5 + p (float): probability of the image being sharpened. Default value is 0.5 """ def __init__(self, sharpness_factor, p=0.5):