RandomVerticalFlip

class torchvision.transforms.v2.RandomVerticalFlip(p: float = 0.5)

Vertically flip the input with a given probability.

If the input is a torch.Tensor or a TVTensor (e.g. Image, Video, BoundingBoxes etc.) it can have arbitrary number of leading batch dimensions. For example, the image can have [..., C, H, W] shape. A bounding box can have [..., 4] shape.

Parameters:

p (float, optional) – probability of the input being flipped. Default value is 0.5

Examples using RandomVerticalFlip:

Illustration of transforms

Illustration of transforms

transform(inpt: Any, params: Dict[str, Any]) → Any

Method to override for custom transforms.

See How to write your own v2 transforms