ElasticTransform¶
- class torchvision.transforms.ElasticTransform(alpha=50.0, sigma=5.0, interpolation=InterpolationMode.BILINEAR, fill=0)[source]¶
对张量图像进行弹性变换。给定 alpha 和 sigma 参数,它将根据随机偏移生成所有像素的位移向量。Alpha 控制位移的强度,sigma 控制位移的平滑度。这些位移被添加到一个恒等网格中,然后使用生成的网格对图像进行 grid_sample 操作。
- 应用
随机变换图像中对象的形态,产生类似透过水看东西的效果。
- 参数:
alpha (float 或 sequence of python:floats) – 位移的幅度。默认值为 50.0。
sigma (float 或 sequence of 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
的示例