transforms.RandomVerticalFlip

最新推荐文章于 2024-01-05 23:50:17 发布

原创最新推荐文章于 2024-01-05 23:50:17 发布 · 959 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #深度学习

深度学习专栏收录该内容

28 篇文章

订阅专栏

transforms.RandomVerticalFlip是一个在PyTorch的torchvision库中用于图像预处理的函数，它以指定概率p（默认0.5）随机将图像垂直翻转。这个函数适用于PILImage或Tensor格式的图像，当处理Tensor时，要求其形状为[C,H,W]。实例程序展示了如何使用该函数对图像进行翻转并展示结果。

PyTorch 2.9

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理

transforms.RandomVerticalFlip()

类原型：

CLASStorchvision.transforms.RandomVerticalFlip(p=0.5)
    
作用：
    以给定的概率随机垂直翻转给定的图像（PIL Image or Tensor）。如果图像是torch张量，则期望它具有[C，H, W]形状，其中C表示任意数量的张量维度
    
参数：
    p (float) ：图像被翻转的概率。默认值为0.5

实例程序：

img = Image.open("data/val/Dog/2.jpg")
print(np.array(img).shape)

randomverticalflip = transforms.RandomVerticalFlip(p=0.5)
img_to = randomverticalflip(img)
img_to.show()