Resize

class torchvision.transforms.Resize(size, interpolation=InterpolationMode.BILINEAR, max_size=None, antialias=True)

將輸入影像調整為給定尺寸。如果影像是 torch Tensor，則預期其形狀為 […, H, W]，其中 … 表示最多兩個前導維度

引數:

• size (sequence 或 int) –

  期望的輸出尺寸。如果 size 是一個序列，如 (h, w)，輸出尺寸將匹配此序列。如果 size 是一個 int，影像的較短邊將匹配此數字。即，如果 height > width，則影像將縮放到 (size * height / width, size)。

  注意

  在 torchscript 模式下不支援 size 為單個 int，請使用長度為 1 的序列：[size, ]。

• interpolation (InterpolationMode) – 由 torchvision.transforms.InterpolationMode 定義的期望插值列舉。預設為 InterpolationMode.BILINEAR。如果輸入是 Tensor，僅支援 InterpolationMode.NEAREST、InterpolationMode.NEAREST_EXACT、InterpolationMode.BILINEAR 和 InterpolationMode.BICUBIC。也接受相應的 Pillow 整數常量，例如 PIL.Image.BILINEAR。

• max_size (int, 可選) – 調整大小後圖像較長邊的最大允許尺寸。如果影像較長邊在根據 size 調整大小後大於 max_size，則 size 將被覆蓋，使較長邊等於 max_size。結果是，較短邊可能小於 size。這僅在 size 是 int (或在 torchscript 模式下是長度為 1 的序列) 時支援。

• antialias (bool, 可選) –

  是否應用抗鋸齒。它僅影響具有雙線性或雙三次模式的 張量，否則將被忽略：對於 PIL 影像，在雙線性或雙三次模式下始終應用抗鋸齒；在其他模式下（對於 PIL 影像和張量），抗鋸齒沒有意義，此引數將被忽略。可能的值為

  • True (預設)：將對抗雙線性或雙三次模式應用抗鋸齒。其他模式不受影響。這可能是您想要使用的設定。

  • False：將不對任何模式下的張量應用抗鋸齒。PIL 影像在雙線性或雙三次模式下仍會應用抗鋸齒，因為 PIL 不支援無抗鋸齒。

  • None：等同於張量的 False 和 PIL 影像的 True。此值是為了相容舊版本而存在，除非您確實知道自己在做什麼，否則您可能不想使用它。

  預設值從 v0.17 中的 None 更改為 True，以使 PIL 和 Tensor 後端保持一致。

使用 Resize 的示例

變換示例

變換示例

forward(img)

引數:

img (PIL Image 或 Tensor) – 要縮放的影像。

返回:

縮放後的影像。

返回型別:

PIL Image 或 Tensor