快捷方式

ElasticTransform

class torchvision.transforms.ElasticTransform(alpha=50.0, sigma=5.0, interpolation=InterpolationMode.BILINEAR, fill=0)[source]

对张量图像进行弹性变换。给定 alpha 和 sigma 参数,它将根据随机偏移生成所有像素的位移向量。Alpha 控制位移的强度,sigma 控制位移的平滑度。这些位移被添加到一个恒等网格中,然后使用生成的网格对图像进行 grid_sample 操作。

应用

随机变换图像中对象的形态,产生类似透过水看东西的效果。

参数:
  • alpha (floatsequence of python:floats) – 位移的幅度。默认值为 50.0。

  • sigma (floatsequence of python:floats) – 位移的平滑度。默认值为 5.0。

  • interpolation (InterpolationMode) – 由 torchvision.transforms.InterpolationMode 定义的期望插值枚举。默认值为 InterpolationMode.BILINEAR。如果输入是 Tensor,仅支持 InterpolationMode.NEARESTInterpolationMode.BILINEAR。也接受相应的 Pillow 整数常量,例如 PIL.Image.BILINEAR

  • fill (sequencenumber) – 变换后图像外部区域的像素填充值。默认值为 0。如果给定一个数字,该值将分别用于所有通道。

使用 ElasticTransform 的示例

变换示意图

变换示意图
forward(tensor: Tensor) Tensor[source]
参数:

tensor (PIL ImageTensor) – 要转换的图像。

返回值:

变换后的图像。

返回类型:

PIL Image 或 Tensor

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

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

查看资源