AugMix

class torchvision.transforms.AugMix(severity: int = 3, mixture_width: int = 3, chain_depth: int = - 1, alpha: float = 1.0, all_ops: bool = True, interpolation: InterpolationMode = InterpolationMode.BILINEAR, fill: Optional[List[float]] = None)

基于 “AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty” 的 AugMix 数据增强方法。如果图像是 torch Tensor，它应该是 torch.uint8 类型，并且预期形状为 [..., 1 或 3, H, W]，其中 ... 表示任意数量的前导维度。如果 img 是 PIL Image，预期模式为 “L” 或 “RGB”。

参数:

• severity (int) – 基本增强操作的强度。默认值为 3。

• mixture_width (int) – 增强链的数量。默认值为 3。

• chain_depth (int) – 增强链的深度。负值表示从区间 [1, 3] 中随机采样的深度。默认值为 -1。

• alpha (float) – 概率分布的超参数。默认值为 1.0。

• all_ops (bool) – 使用所有操作（包括亮度、对比度、色彩和锐度）。默认值为 True。

• interpolation (InterpolationMode) – 由 torchvision.transforms.InterpolationMode 定义的期望插值枚举。默认值为 InterpolationMode.NEAREST。如果输入为 Tensor，仅支持 InterpolationMode.NEAREST 和 InterpolationMode.BILINEAR。

• fill (sequence or number, optional) – 转换后图像区域外的像素填充值。如果给定一个数字，则该值将用于所有波段。

使用 AugMix 的示例

转换图示

转换图示

forward(orig_img: Tensor) → Tensor

img (PIL Image 或 Tensor)：待转换图像。

返回值:

转换后的图像。

返回类型:

PIL Image 或 Tensor