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