调整大小¶
- class torchvision.transforms.Resize(size, interpolation=InterpolationMode.BILINEAR, max_size=None, antialias=True)[源代码]¶
将输入图像的大小调整为给定的大小。如果图像是 torch Tensor,则预期其形状为 […, H, W],其中 … 表示最多两个前导维度
- 参数:
size (sequence 或 int) –
所需的输出大小。如果 size 是 (h, w) 这样的序列,则输出大小将与之匹配。如果 size 是一个整数,图像的较短边将匹配此数字。即,如果 height > width,则图像将缩放到 (size * height / width, size)。
注意
在 torchscript 模式下,不支持将 size 作为单个整数,请使用长度为 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) –
是否应用抗锯齿。它仅影响双线性或双三次模式下的张量,否则将被忽略:在 PIL 图像上,双线性或双三次模式始终应用抗锯齿;在其他模式下(对于 PIL 图像和张量),抗锯齿没有意义,此参数将被忽略。可能的值为
True(默认):将对双线性或双三次模式应用抗锯齿。其他模式不受影响。这可能是您想要使用的。False:将不对任何模式下的张量应用抗锯齿。PIL 图像在双线性或双三次模式下仍然进行抗锯齿处理,因为 PIL 不支持无抗锯齿。None:对于张量相当于False,对于 PIL 图像相当于True。此值存在是为了兼容性,除非您真的知道自己在做什么,否则可能不希望使用它。
默认值在 v0.17 中从
None更改为True,以使 PIL 和 Tensor 后端保持一致。
使用
Resize的示例