快捷方式

随机擦除

class torchvision.transforms.v2.RandomErasing(p: float = 0.5, scale: Sequence[float] = (0.02, 0.33), ratio: Sequence[float] = (0.3, 3.3), value: float = 0.0, inplace: bool = False)[source]

随机选择输入图像或视频中的矩形区域并擦除其像素。

此转换不支持 PIL Image。Zhong 等人的“随机擦除数据增强”。查看 https://arxiv.org/abs/1708.04896

参数:
  • p (float, 可选) – 执行随机擦除操作的概率。

  • scale (python:float 元组, 可选) – 擦除区域相对于输入图像的比例范围。

  • ratio (python:float 元组, 可选) – 擦除区域的纵横比范围。

  • value (数字数字元组) – 擦除值。默认值为 0。如果为单个整数,则用于擦除所有像素。如果为长度为 3 的元组,则分别用于擦除 R、G、B 通道。如果为“随机”字符串,则用随机值擦除每个像素。

  • inplace (bool, 可选) – 是否将此转换设置为就地转换。默认设置为 False。

返回值:

擦除的输入。

示例

>>> from torchvision.transforms import v2 as transforms
>>>
>>> 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(),
>>> ])
static get_params(img: Tensor, scale: Tuple[float, float], ratio: Tuple[float, float], value: Optional[List[float]] = None) Tuple[int, int, int, int, Tensor][source]

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

参数:
  • img (Tensor) – 要擦除的张量图像。

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

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

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

返回值:

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

返回类型:

元组

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题解答

查看资源