Python PyTorch RandomErasing用法及代碼示例

本文簡要介紹python語言中 torchvision.transforms.RandomErasing 的用法。

用法:

class torchvision.transforms.RandomErasing(p=0.5, scale=(0.02, 0.33), ratio=(0.3, 3.3), value=0, inplace=False)

參數：

• p-執行隨機擦除操作的概率。

• scale-擦除區域與輸入圖像的比例範圍。

• ratio-擦除區域的縱橫比範圍。

• value-擦除值。默認值為 0。如果是單個 int，則用於擦除所有像素。如果是長度為3的元組，則分別用於擦除R、G、B通道。如果是 ‘random’ 的 str，則使用隨機值擦除每個像素。

• inplace-布爾值以使此轉換就地。默認設置為 False。

返回：

刪除的圖像。

在火炬張量圖像中隨機選擇一個矩形區域並擦除其像素。此轉換不支持 PIL 圖像。鍾等人的“隨機擦除數據增強”。參見https://arxiv.org/abs/1708.04896

示例

>>> transform = transforms.Compose([
>>>   transforms.RandomHorizontalFlip(),
>>>   transforms.PILToTensor(),
>>>   transforms.ConvertImageDtype(torch.float),
>>>   transforms.Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225)),
>>>   transforms.RandomErasing(),
>>> ])