快捷方式

ElasticTransform

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

使用弹性变换转换张量图像。给定 alpha 和 sigma,它将基于随机偏移为所有像素生成位移向量。Alpha 控制强度,sigma 控制位移的平滑度。位移被添加到单位网格中,生成的网格用于从图像中进行网格采样。

应用

随机变换图像中物体的形态,并产生类似于透过水观看的效果。

参数:
  • alpha (floatpython:floats 序列) – 位移的大小。默认为 50.0。

  • sigma (floatpython:floats 序列) – 位移的平滑度。默认为 5.0。

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

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

使用 ElasticTransform 的示例

变换图示

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

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

返回值:

变换后的图像。

返回类型:

PIL Image 或 Tensor

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源