AugMix¶
- class torchvision.transforms.v2.AugMix(severity: int = 3, mixture_width: int = 3, chain_depth: int = - 1, alpha: float = 1.0, all_ops: bool = True, interpolation: Union[InterpolationMode, int] = InterpolationMode.BILINEAR, fill: Union[int, float, Sequence[int], Sequence[float], None, dict[Union[type, str], Union[int, float, collections.abc.Sequence[int], collections.abc.Sequence[float], NoneType]] = None)[源代码]¶
AugMix 数据增强方法，基于 “AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty”。
此转换仅适用于图像和视频。
如果输入是
torch.Tensor，则其类型应为torch.uint8，并且期望其形状为 […, 1 或 3, H, W]，其中 … 表示任意数量的前导维度。如果 img 是 PIL Image，则期望其模式为“L”或“RGB”。- 参数:
severity (int, optional) – 基本增强算子的强度。默认为
3。mixture_width (int, optional) – 增强链的数量。默认为
3。chain_depth (int, optional) – 增强链的深度。负值表示从区间 [1, 3] 中随机采样。默认为
-1。alpha (float, optional) – 概率分布的超参数。默认为
1.0。all_ops (bool, optional) – 使用所有操作（包括亮度、对比度、色彩和锐度）。默认为
True。interpolation (InterpolationMode, optional) – 由
torchvision.transforms.InterpolationMode定义的期望的插值枚举。默认为InterpolationMode.NEAREST。如果输入是 Tensor，则只支持InterpolationMode.NEAREST和InterpolationMode.BILINEAR。fill (sequence 或 number, 可选) – 转换后图像外部区域的像素填充值。如果给定一个数字，则该值分别用于所有通道。
使用
AugMix的示例
