RandomInvert — Torchvision 0.22 documentation (original) (raw)
class torchvision.transforms.v2.RandomInvert(p: float = 0.5)[source]¶
Inverts the colors of the given image or video with a given probability.
If img is a Tensor, it is expected to be in […, 1 or 3, H, W] format, where … means it can have an arbitrary number of leading dimensions. If img is PIL Image, it is expected to be in mode “L” or “RGB”.
Parameters:
p (float) – probability of the image being color inverted. Default value is 0.5
Examples using RandomInvert:
transform(inpt: Any, params: Dict[str, Any]) → Any[source]¶
Method to override for custom transforms.