RandomErasing

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

在 torch.Tensor 图像中随机选择一个矩形区域并擦除其像素。此变换不支持 PIL Image。该方法基于 Zhong 等人的论文“Random Erasing Data Augmentation”。参见 https://arxiv.org/abs/1708.04896

参数:

• p – 执行随机擦除操作的概率。

• scale – 擦除区域相对于输入图像的比例范围。

• ratio – 擦除区域的纵横比范围。

• value – 擦除值。默认为 0。如果是单个 int，则用于擦除所有像素。如果是长度为 3 的元组，则分别用于擦除 R、G、B 通道。如果是字符串 ‘random’，则用随机值擦除每个像素。

• inplace – 布尔值，指示是否进行原地（inplace）变换。默认为 False。

返回值:

擦除后的图像。

示例

>>> 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(),
>>> ])

forward(img)

参数:

img (Tensor) – 待擦除的 Tensor 图像。

返回值:

擦除后的 Tensor 图像。

返回类型:

img (Tensor)

static get_params(img: Tensor, scale: Tuple[float, float], ratio: Tuple[float, float], value: Optional[List[float]] = None) → Tuple[int, int, int, int, Tensor]

获取随机擦除的参数，用于 erase 函数。

参数:

• img (Tensor) – 待擦除的 Tensor 图像。

• scale (sequence) – 擦除区域相对于输入图像的比例范围。

• ratio (sequence) – 擦除区域的纵横比范围。

• value (列表, 可选) – 擦除值。如果为 None，则解释为“随机”（用随机值擦除每个像素）。如果 len(value) 为 1，则解释为一个数字，即 value[0]。

返回值:

参数 (i, j, h, w, v)，将传递给 erase 函数用于随机擦除。

返回类型:

元组