ElasticTransform

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

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

应用

随机变换图像中物体的形态，并产生类似看穿水面的效果。

参数:

• alpha (float 或 python:floats 序列) – 位移的大小。默认为 50.0。

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

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

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

使用 ElasticTransform 的示例

变换的图示

变换的图示

forward(tensor: Tensor) → Tensor

参数:

tensor (PIL Image 或 Tensor) – 要变换的图像。

返回值:

变换后的图像。

返回类型:

PIL Image 或 Tensor