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 or 3, H, W] 形狀，其中 … 表示任意數量的領先維度。如果 img 是 PIL 影像，則預期為 “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