随机水平翻转¶
- class torchvision.transforms.v2.RandomHorizontalFlip(p: float = 0.5)[source]¶
以给定概率水平翻转输入。
如果输入是
torch.Tensor
或TVTensor
(例如Image
、Video
、BoundingBoxes
等),它可以具有任意数量的前导批处理维度。例如,图像可以具有[..., C, H, W]
形状。边界框可以具有[..., 4]
形状。- 参数:
p (float, 可选) – 输入被翻转的概率。默认值为 0.5
使用
RandomHorizontalFlip
的示例Transforms v2 入门Transforms 的说明Transforms v2:端到端目标检测/分割示例如何使用 CutMix 和 MixUp如何编写自己的 v2 Transforms如何编写自己的 TVTensor 类