ElasticTransform

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

對張量影像進行彈性變換。給定 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 的示例

變換示意圖

變換示意圖

forward(tensor: Tensor) → Tensor

引數：

tensor (PIL Image 或 Tensor) – 要轉換的影像。

返回值：

變換後的影像。

返回型別：

PIL Image 或 Tensor