### 关于 `transforms.Flip()` 的说明 `transforms.Flip()` 并不是 PyTorch 中标准的图像变换类之一。通常情况下，对于翻转操作，PyTorch 提供了更具体的类来实现水平或垂直方向上的随机翻转效果。 #### 实现水平和垂直翻转的方法 为了达到类似于 `Flip()` 的功能，可以分别使用： - **水平翻转** 使用 `RandomHorizontalFlip` 类来进行依概率 p 的水平翻转操作[^3]。 ```python from torchvision import transforms transform_horizontal = transforms.Compose([ transforms.RandomHorizontalFlip(p=1.0) # 设置p=1表示总是执行此操作 ]) ``` - **垂直翻转** 同样地，利用 `RandomVerticalFlip` 来完成垂直方向上的翻转。 ```python transform_vertical = transforms.Compose([ transforms.RandomVerticalFlip(p=1.0) # 设置p=1表示总是执行此操作 ]) ``` 上述两个例子展示了如何通过设置参数 `p` 控制翻转发生的几率；当 `p=1.0` 时意味着每次都会应用该变换。 如果确实需要一个通用的翻转接口而不仅仅是随机性的，则可以通过自定义方式组合这两个基本组件或者编写自己的转换逻辑。 #### 自定义 Flip 转换器 考虑到灵活性的需求，也可以创建一个新的转换器来自定义行为: ```python from torchvision.transforms import functional as F import random class CustomFlip(object): """Custom flip transformation that can perform horizontal and/or vertical flips.""" def __init__(self, hflip_prob=0.5, vflip_prob=0.5): self.hflip_prob = hflip_prob self.vflip_prob = vflip_prob def __call__(self, img): if random.random() < self.hflip_prob: img = F.hflip(img) if random.random() < self.vflip_prob: img = F.vflip(img) return img ``` 这段代码实现了可以根据给定的概率独立决定是否进行水平 (`hflip`) 和垂直 (`vflip`) 方向上的翻转的功能。