首页 > 解决方案 > pytorchalbumentations 增强 p 值?

问题描述

augmented_images(raw.image_id.unique()[1230], albumentations.HorizontalFlip(p=1))

对于 augmented_image,p=1 是什么意思?价值差异使角度不同吗?如果不是,我应该如何进行各种角度不同的水平增强?

标签: pytorchtorchtorchvisionalbumentations

解决方案


正如您在以下文档中看到的albumentations.HorizontalFlip

参数: p (float) – 应用变换的概率。默认值:0.5。

如果你想旋转,你应该考虑使用albumentations.augmentations.transforms.Rotate

Rotate(limit=90, interpolation=1, border_mode=4, value=None, mask_value=None, always_apply=False, p=0.5)

将输入旋转一个从均匀分布中随机选择的角度。

参数:

  • limit ((int, int) or int) - 随机选择角度的范围。如果 limit 是单个 int,则从 (-limit, limit) 中选择一个角度。默认值:(-90, 90)
  • [...]
  • p (float) – 应用变换的概率。默认值:0.5。

推荐阅读