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], Optional[Union[int, float, Sequence[int], Sequence[float]]]]] = None)[source]¶
基於"AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty"的 AugMix 資料增強方法。
此轉換僅適用於圖片和影片。
如果輸入是
torch.Tensor，則其類型應為torch.uint8，並且預期具有 […, 1 or 3, H, W] 形狀，其中 … 表示任意數量的leading dimensions。如果 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 or number, optional) – 轉換後圖像外部區域的像素填充值。如果給定一個數字，則該值分別用於所有通道。
使用
AugMix的範例
