調整大小

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

將輸入圖像調整為給定的大小。如果圖像為 torch Tensor，則預期具有 […, H, W] 形狀，其中 … 表示最多兩個前導維度

參數:

• size (序列 或 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, optional) – 調整大小後的影像，其較長邊所允許的最大長度。如果根據 size 調整大小後，影像的較長邊大於 max_size，則 size 將被覆寫，使較長邊等於 max_size。因此，較短邊可能會短於 size。這僅在 size 為整數（或在 torchscript 模式下長度為 1 的序列）時才支援。

• antialias (bool, optional) –

  是否應用抗鋸齒。這只會影響具有雙線性或雙立方模式的 tensors，否則會被忽略：在 PIL 影像上，始終會在雙線性或雙立方模式上應用抗鋸齒；在其他模式（對於 PIL 影像和 tensors）上，抗鋸齒沒有意義，因此會忽略此參數。可能的值為

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

  • False：將不會對任何模式下的 tensors 應用抗鋸齒。PIL 影像仍然會在雙線性或雙立方模式上進行抗鋸齒處理，因為 PIL 不支援不進行抗鋸齒。

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

  為了讓 PIL 和 Tensor 後端保持一致，預設值已從 v0.17 中的 None 變更為 True。

使用 Resize 的範例

transforms 的圖解

transforms 的圖解

影片 API

影片 API

forward(img)

參數:

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

傳回:

重新縮放的影像。

傳回類型:

PIL Image 或 Tensor