RandomHorizontalFlip¶
- class torchvision.transforms.v2.RandomHorizontalFlip(p: float = 0.5)[源码]¶
以给定的概率水平翻转输入。
如果输入是
torch.Tensor
或TVTensor
(例如Image
、Video
、BoundingBoxes
等),它可以包含任意数量的前导批处理维度。例如,图像可以具有[..., C, H, W]
的形状。边界框可以具有[..., 4]
的形状。- 参数:
p (float, 可选) – 输入被翻转的概率。默认值为 0.5
使用
RandomHorizontalFlip
的示例