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 資料增強方法，基於 “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) – 基本增強運算元的嚴重程度。預設為 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 或 number, 可選) – 轉換後圖像外部區域的畫素填充值。如果給定一個數字，則該值分別用於所有通道。

使用 AugMix 的示例

轉換圖示

轉換圖示

forward(orig_img: Tensor) → Tensor

img (PIL Image 或 Tensor): 要轉換的影像。

返回:

轉換後的影像。

返回型別:

PIL 影像或張量