CenterCrop¶
- class torchvision.transforms.CenterCrop(size)[源代码]¶
将给定图像在中心裁剪。如果图像是 torch Tensor,则期望其形状为 […, H, W],其中 … 表示任意数量的前导维度。如果图像在任何边缘的尺寸小于输出尺寸,则图像将用 0 填充然后中心裁剪。
- 参数:
size (sequence 或 int) – 裁剪的期望输出尺寸。如果 size 是一个 int 而不是像 (h, w) 这样的序列,则会进行一个正方形裁剪 (size, size)。如果提供长度为 1 的序列,则将其解释为 (size[0], size[0])。
使用
CenterCrop
的示例