RandomHorizontalFlip — Torchvision 0.22 documentation (original) (raw)
class torchvision.transforms.v2.RandomHorizontalFlip(p: float = 0.5)[source]¶
Horizontally 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 RandomHorizontalFlip:
transform(inpt: Any, params: Dict[str, Any]) → Any[source]¶
Method to override for custom transforms.